Search papers, labs, and topics across Lattice.
The paper introduces PIME, a novel framework for fMRI-based disorder diagnosis that combines prototype-based classification and consistency training with structural perturbations to learn a structured latent space. PIME uses Monte Carlo Tree Search (MCTS) under a prototype-consistent objective to extract compact, minimal-sufficient explanatory subgraphs post-training, addressing issues of noisy interactions and unreliable post-hoc attribution methods in standard deep learning approaches. Experiments on three fMRI datasets demonstrate state-of-the-art performance, high reproducibility (90%), and identification of critical brain regions consistent with neuroimaging findings.
PIME leverages prototype-guided Monte Carlo Tree Search to extract compact, neuroscientifically-validated brain subnetworks predictive of disorder, outperforming standard deep learning approaches in both accuracy and interpretability.
Recent deep learning methods for fMRI-based diagnosis have achieved promising accuracy by modeling functional connectivity networks. However, standard approaches often struggle with noisy interactions, and conventional post-hoc attribution methods may lack reliability, potentially highlighting dataset-specific artifacts. To address these challenges, we introduce PIME, an interpretable framework that bridges intrinsic interpretability with minimal-sufficient subgraph optimization by integrating prototype-based classification and consistency training with structural perturbations during learning. This encourages a structured latent space and enables Monte Carlo Tree Search (MCTS) under a prototype-consistent objective to extract compact minimal-sufficient explanatory subgraphs post-training. Experiments on three benchmark fMRI datasets demonstrate that PIME achieves state-of-the-art performance. Furthermore, by constraining the search space via learned prototypes, PIME identifies critical brain regions that are consistent with established neuroimaging findings. Stability analysis shows 90% reproducibility and consistent explanations across atlases.
