Accepted for ISBI 2023