Search papers, labs, and topics across Lattice.
This paper introduces "resilience," a novel uncertainty estimation method for neural cellular automata (NCAs) used in medical image segmentation. Resilience measures the stability of NCA predictions by perturbing the automaton state and observing if it converges back to the original solution, with stable predictions considered confident. Experiments on medical segmentation benchmarks demonstrate that resilience outperforms baselines in identifying failure cases, as measured by selective prediction (螖Dice@90, AURC) and ranking metrics (AUROC, AUPRC).
A simple "resilience" metric turns out to be surprisingly effective at identifying failure cases in NCA-based medical image segmentation, improving trust without retraining.
Neural cellular automata (NCA) provide a lightweight alternative to encoder-decoder segmentation networks. However, it can be difficult to decide when a prediction should be trusted. Here, we study uncertainty estimation for NCA-based medical image segmentation without modifying the underlying architecture or retraining the model. Our approach is motivated by viewing the NCA as a dynamical system where convergent attractors correspond to confident predictions. Concretely, we propose resilience, a simple measure that leverages the intrinsic iterative structure of NCAs by probing the stability of the final prediction under small perturbations of the automaton state. Predictions that return to the same solution are deemed confident, while those that change substantially are flagged as uncertain. We evaluate uncertainty by its ability to predict segmentation quality using selective prediction metrics ($螖$Dice@90 and AURC) and ranking metrics (AUROC and AUPRC). Across multiple medical segmentation benchmarks, resilience identifies failure cases more reliably than baselines, improving trust and safety in NCA-based models.
