Augmented Equivariant Mesh Networks for Anatomical Mesh Segmentation

Daniel Saragih

Additional Submission Instructions: For the camera-ready version, please include the author names and affiliations, funding disclosures, and acknowledgements.

Track: Track 1: Original Research/Position/Education/Attention Track

Keywords: anatomical mesh segmentation, equivariant graph networks, robustness, clinical data

TL;DR: We present EAMS, a lightweight equivariant mesh segmentation framework augmented with anatomy-aware intrinsic features and long-range context modules.

Abstract: Anatomical mesh segmentation requires models that operate directly on irregular surface geometry while remaining robust to arbitrary patient pose and mesh resolution variation. Existing task-specific mesh and point-cloud methods are not equivariant, and can degrade sharply under test-time perturbation, for example dropping by 25-26 IoU points on intraoral scan segmentation at 40$\textdegree{}$ tilt. We present EAMS, an Equivariant Anatomical Mesh Segmentor built on Equivariant Mesh Neural Networks (EMNN), and evaluate it across four clinically distinct tasks spanning edge-, vertex-, and face-level supervision. We combine intrinsic mesh descriptors with anatomy-aware priors, including PCA-derived frames for dental arches and liver surfaces, and augment message passing with soft regional aggregators and virtual nodes to capture multi-scale geometric context. Across intracranial aneurysm and intraoral segmentation, EAMS variants are competitive with specialized baselines on unperturbed inputs while remaining stable under geometric perturbations, and on liver surfaces they expose a favorable trade-off between canonical-pose accuracy and rotation robustness. These results position equivariant mesh models as a useful substrate for assistive scientific workflows on geometric biomedical data, where reproducible behavior under arbitrary orientation matters as much as raw accuracy.

Submission Number: 268