FDIF: Formula-Driven supervised Learning with Implicit Functions for 3D Medical Image Segmentation
Organizations: Waseda University · National Institute of Advanced Industrial Science and Technology · The University of Osaka · University of Oxford
Abstract
Deep learning-based 3D medical image segmentation methods relies on large-scale labeled datasets, yet acquiring such data is difficult due to privacy constraints and the high cost of expert annotation. Formula-Driven Supervised Learning (FDSL) offers an appealing alternative by generating training data and labels directly from mathematical formulas. However, existing voxel-based approaches are limited in geometric expressiveness and cannot synthesize realistic textures. We introduce Formula-Driven supervised learning with Implicit Functions (FDIF), a framework that enables scalable pre-training without using any real data and medical expert annotations. FDIF introduces an implicit-function representation based on signed distance functions (SDFs), enabling compact modeling of complex geometries while exploiting the surface representation of SDFs to support controllable synthesis of both geometric and intensity textures. Across three medical image segmentation benchmarks (AMOS, ACDC, and KiTS) and three architectures (SwinUNETR, nnUNet ResEnc-L, and nnUNet Primus-M), FDIF consistently improves over a formula-driven method, and achieves performance comparable to self-supervised approaches pre-trained on large-scale real datasets. We further show that FDIF pre-training also benefits 3D classification tasks, highlighting implicit-function-based formula supervision as a promising paradigm for data-free representation learning. Code is available at https://github.com/yamanoko/FDIF.
Figures & tables
| Architecture | Pre-training | Type | AMOS | ACDC | KiTS | Avg |
| SwinUNETR | Scratch | - | 80.98 | 81.91 | 74.55 | 79.15 |
| PrimGeoSeg [ 23 ] | FDSL | 81.89 | 85.02 | 75.80 | 80.90 | |
| Ours (FDIF, Disp+Map) | 82.28 | 83.82 | 77.20 | 81.10 | ||
| Ours (FDIF, Map only) | 82.70 | 84.66 | 78.31 | 81.89 | ||
| nnUNet ResEnc-L | Scratch | - | 85.97 | 91.98 | 83.84 | 87.26 |
| S3D [ 27 , 28 ] | SSL | 86.16 | 92.01 | 86.01 | 88.06 |
| Pre-training | Avg | Spl | RKd | LKd | Gal | Eso | Liv | Sto | Aor | Pos | Pan | RAd | LAd | Duo | Bla | P/U |
| 10 global-shape | 87.95 | 96.84 | 96.00 | 95.58 | 83.32 | 82.59 | 97.65 | 91.17 | 94.91 | 89.87 | 87.15 | 75.90 | 77.27 | 81.09 | 86.90 | 82.95 |
| 109 global-shape | 87.96 | 96.89 | 96.20 | 95.03 | 84.36 | 83.03 | 97.67 | 90.80 | 94.68 | 90.13 | 87.09 | 76.11 | 77.38 | 80.89 | 87.11 | 82.01 |
| 109 gs+10 Disp | 88.02 | 96.77 | 95.96 | 95.49 | 84.66 | 82.79 | 97.68 | 91.10 | 94.87 | 90.24 | 87.34 | 75.50 | 77.62 | 81.51 | 86.77 | 82.00 |
| 109 gs+10 Map | 88.04 | 96.86 | 96.14 | 95.47 | 84.19 | 83.00 | 97.69 | 90.93 | 94.87 | 90.21 | 87.49 | 76.07 | 77.47 | 81.53 | 86.24 | 82.45 |
| 109 gs+10 Disp+10 Map | 87.66 | 96.78 | 96.06 | 95.55 | 84.11 | 82.53 | 97.69 | 90.58 | 94.82 | 90.24 | 87.46 | 76.06 | 77.41 | 81.59 | 81.15 | 82.89 |
| PrimGeoSeg (32 gs) | 87.57 | 96.80 | 95.76 | 95.44 | 83.20 | 81.96 | 97.56 | 90.37 | 94.78 | 89.96 | 87.04 | 75.85 | 76.89 | 80.36 | 85.62 | 82.01 |
| Pre-training | Avg | Spl | RKd | LKd | Gal | Eso | Liv | Sto | Aor | Pos | Pan | RAd | LAd | Duo | Bla | P/U |
| 10 gs (Extrusion only) | 87.80 | 96.93 | 95.82 | 94.90 | 83.31 | 82.37 | 97.62 | 90.49 | 94.86 | 89.79 | 87.01 | 75.88 | 77.14 | 81.13 | 86.34 | 83.36 |
| 10 gs (Revolution & Hollowing only) | 87.78 | 96.85 | 96.20 | 95.42 | 83.92 | 82.24 | 97.62 | 90.93 | 94.76 | 89.86 | 87.20 | 76.25 | 76.99 | 81.03 | 85.40 | 82.07 |
| 10 gs (Combined) | 87.95 | 96.84 | 96.00 | 95.58 | 83.32 | 82.59 | 97.65 | 91.17 | 94.91 | 89.87 | 87.15 | 75.90 | 77.27 | 81.09 | 86.90 | 82.95 |
| Pre-training | MRNet | MedMNIST Organ | MedMNIST Nodule | MedMNIST Fracture | Avg |
| Ours (FDIF) | 67.45 | 96.52 | 79.72 | 69.21 | 78.23 |
| SwinUNETR (SSL) [ 25 ] | 67.36 | 92.05 | 80.48 | 67.75 | 76.91 |
| Scratch | 62.19 | 85.32 | 85.12 | 56.22 | 72.21 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
3D Medical Image Segmentation
3D Medical Image Segmentation identifies organs, lesions, and fine structures in volumetric scans for measurement and clinical planning.
See all 212 papers