arXiv:2503.20326eess.IV2025-03中稿 · MICCAI 2025 PRIME …被引 2

让脑病变分割模型像医生一样持续学习新模态和病种,同时保护数据隐私。

Modality-Agnostic Brain Lesion Segmentation with Privacy-aware Continual Learning

  • 用混合专家与双知识蒸馏,实现多数据集渐进式学习。
  • 在5个数据集上平均提升14%的分割准确率(Dice)。
  • 适合需要跨医院、多模态、持续更新的医学影像系统。

传统脑病变分割模型通常针对特定病理,依赖预定义模态的数据集。面对新模态或新病种时,常需重新训练独立模型,与医生循序渐进积累经验的方式相悖。受此启发,我们提出一种统一分割模型,可依次从包含不同模态和病种的数据集中学习。方法采用隐私感知的持续学习框架,结合混合专家机制与双重知识蒸馏,有效缓解灾难性遗忘,同时不损害对新数据集的表现。在五个多样化脑MRI数据集及四种数据序列上的实验表明,该框架能维持单一可适配模型,应对不同医院协议、成像模态和疾病类型。相比广泛使用的隐私感知持续学习方法(如LwF、SI、EWC、MiB、TED),本方法平均Dice分数提升约14%。该框架为更灵活实用的脑病变分割模型迈出了重要一步,代码已开源于GitHub。

原文摘要 · Abstract (English)

Traditional brain lesion segmentation models for multi-modal MRI are typically tailored to specific pathologies, relying on datasets with predefined modalities. Adapting to new MRI modalities or pathologies often requires training separate models, which contrasts with how medical professionals incrementally expand their expertise by learning from diverse datasets over time. Inspired by this human learning process, we propose a unified segmentation model capable of sequentially learning from multiple datasets with varying modalities and pathologies. Our approach leverages a privacy-aware continual learning framework that integrates a mixture-of-experts mechanism and dual knowledge distillation to mitigate catastrophic forgetting while not compromising performance on newly encountered datasets. Extensive experiments across five diverse brain MRI datasets and four dataset sequences demonstrate the effectiveness of our framework in maintaining a single adaptable model, capable of handling varying hospital protocols, imaging modalities, and disease types. Compared to widely used privacy-aware continual learning methods such as LwF, SI, EWC, MiB, and TED, our method achieves an average Dice score improvement of approximately 14%. Our framework represents a significant step toward more versatile and practical brain lesion segmentation models, with implementation available on \href{https://github.com/xmindflow/BrainCL}{GitHub}.

脑病变分割持续学习隐私保护多模态

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。