A 211K Parameter Foundation Model: Genetic Prompting in Neural Cellular Automata
Organizations: Darmstadt University of Technology, Karolinenplatz 5, 64289 Darmstadt, Germany
Abstract
The scale of modern foundation models enables broad task generalization. In medical imaging, where segmentation demands vary widely across clinical contexts, such flexibility is invaluable. But current models, such as SAM trade adaptability for scale, locking their utility behind high-end infrastructure and leaving behind the low-resource environments most in need of diagnostic support. To bridge this gap, we introduce Foundational-NCA, a lightweight foundation model capable of adapting to a wide range of medical segmentation tasks without retraining. Built on Neural Cellular Automata (NCA), Foundational-NCA incorporates gradient-based learning with genetic optimization over a latent task space, using a compact task vector to guide predictions. This enables dynamic adaptation via a genetic prompt mechanism, allowing the model to generalize across tasks while keeping the parameter count magnitudes lower at just 211 thousand, compared to SAM-based models with 38.9 to 636 million parameters. We evaluate Foundational-NCA on 15 trained and 6 entirely unseen segmentation tasks from the TotalSegmentator dataset, achieving an average Dice score of 84.5% on known tasks and 60.4% on unseen ones. These results demonstrate that task generalization is possible in ultra-compact models, laying the groundwork for resource-efficient, deployable foundation models in clinical environments where infrastructure is limited. By reducing model size by over 180 times, Foundational-NCA takes a step toward a more equitable future for medical AI, one where powerful tools are within reach of every healthcare setting.
Figures & tables
| Task Set A | Foundational-NCA | SAM | SAM 2 | Med-SAM |
|---|---|---|---|---|
| Seen | 211K param. | 93.7M param. | 38.9M param. | 93.7M param. |
| Gluteus Max. L | 0.885 0.077 | 0.054 0.014 | 0.073 0.019 | 0.814 0.057 |
| Gluteus Max. R | 0.874 0.073 | 0.076 0.021 | 0.075 0.020 | 0.796 0.064 |
| Liver | 0.911 0.105 | 0.343 0.076 | 0.369 0.092 | 0.891 0.068 |
| Lung Lower Lobe R | 0.832 0.162 | 0.638 0.163 | 0.552 0.174 | 0.809 0.116 |
| Lung Middle Lobe R | 0.808 0.114 | 0.463 0.125 | 0.312 0.109 | 0.842 0.080 |
| Task Set A | Foundational-NCA | No Location | No Task |
|---|---|---|---|
| Average | 0.845 | 0.753 | 0.848 |
| Task Set B | Foundational-NCA | No Location | No Task |
| Autochthon L | 0.678 0.159 | 0.002 0.012 | 0.661 0.148 |
| Iliopsoas L | 0.568 0.174 | 0.020 0.041 | 0.593 0.141 |
| Lung Lower Lobe L | 0.730 0.153 | 0.299 0.168 | 0.746 0.158 |
| Sternum | 0.661 0.088 | 0.018 0.038 | 0.645 0.079 |
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Task Set A | Foundational-NCA | No Location | No Task | Single Stage |
|---|---|---|---|---|
| Gluteus Max. L | 0.885 0.077 | 0.867 0.114 | 0.879 0.085 | 0.879 0.090 |
| Gluteus Max. R | 0.874 0.073 | 0.807 0.131 | 0.876 0.089 | 0.877 0.098 |
| Liver | 0.911 0.105 | 0.874 0.170 | 0.892 0.116 | 0.904 0.122 |
| Lung Lower Lobe R | 0.832 0.162 | 0.763 0.188 | 0.886 0.139 | 0.813 0.173 |
| Lung Middle Lobe R | 0.808 0.114 | 0.736 0.143 | 0.837 0.084 | 0.825 0.106 |
| Lung Upper Lobe L | 0.870 0.170 | 0.786 0.205 | 0.887 0.156 | 0.841 0.186 |
| Task Set A | SAM-H | SAM2.1-L |
|---|---|---|
| Parameters | 641.1M | 224.4M |
| Gluteus Max. L | 0.076 0.020 | 0.074 0.021 |
| Gluteus Max. R | 0.074 0.019 | 0.077 0.022 |
| Liver | 0.367 0.089 | 0.443 0.124 |
| Lung Lower Lobe R | 0.583 0.176 | 0.532 0.171 |
| Lung Middle Lobe R | 0.349 0.122 | 0.305 0.106 |
Medical Imaging Foundation Models
Medical imaging foundation models learn reusable representations for tasks across clinical images, modalities, and institutions.
See all 235 papers