arXiv:2606.29200cs.LG2026-06中稿 · ECCV

用黎曼几何构建脑网络原型,实现无源跨站点精神疾病诊断。

BrainRiem: Riemannian Prototype Learning for Source-Free Cross-Site Brain Network Diagnosis

论文配图:BrainRiem: Riemannian Prototype Learning for Source-Free Cross-Site Brain Network Diagnosis
图 1 · 摘自论文原文
  • 基于黎曼流形优化学习紧凑脑网络原型,避免欧氏空间畸变。
  • 在ABIDE和REST-meta-MDD数据集上超越现有方法,准确率提升3.2%~5.8%。
  • 仅传输匿名原型,保护隐私且可解释性强,适合医疗数据协作。

多中心功能磁共振成像研究对精神疾病稳健诊断至关重要,但受扫描仪差异、人口特征及站点采集协议影响,存在严重领域偏移。传统域适应需同时访问源端与目标端数据,违反临床隐私规范。此外,功能连接矩阵位于对称正定(SPD)流形上,使用欧氏运算会引入几何畸变,破坏诊断模式。本文提出BrainRiem,一种无需源数据的域适应框架,通过流形感知的双层优化学习紧凑的黎曼脑网络原型。采用对数欧氏度量确保原型保持有效的SPD矩阵形式,结合狄利克雷能量频谱校准,使原型频率特性与真实脑网络一致。仅需将匿名原型发送至目标站点,作为训练本地模型的稳定锚点,且在评估攻击下显著降低信息泄露风险。在ABIDE与REST-meta-MDD数据集上的实验表明,BrainRiem在多种扫描仪和人口背景下持续优于最先进的无源、传统及图域适应方法。值得注意的是,学习到的原型展现出与神经科学已有发现一致的生物可解释性连接模式,验证了黎曼几何在脑网络分析中的必要性。

原文摘要 · Abstract (English)

Multi-site functional MRI (fMRI) studies are essential for robust neuropsychiatric diagnosis yet suffer severe domain shifts from scanner heterogeneity, demographics, and site-specific acquisition protocols. Traditional domain adaptation requires concurrent source and target data access, violating clinical privacy regulations. Moreover, functional connectivity matrices lie on the Symmetric Positive Definite (SPD) manifold, where Euclidean operations cause geometric distortions corrupting diagnostic patterns. We propose BrainRiem, a source-free domain adaptation framework learning compact Riemannian brain prototypes via manifold-aware bi-level optimization. It employs the Log-Euclidean Metric to ensure prototypes remain valid SPD matrices, while Dirichlet Energy spectral calibration aligns their frequency characteristics with real brain networks. Only anonymized prototypes are transmitted to target sites, serving as stable anchors for training local models without source data access and reducing leakage under the evaluated attacks. Comprehensive experiments on ABIDE and REST-meta-MDD show BrainRiem consistently outperforms state-of-the-art source-free, traditional, and graph domain adaptation methods across diverse scanners and demographics. Notably, learned prototypes exhibit biologically interpretable connectivity patterns aligning with established neuroscience findings, validating the necessity of Riemannian geometry for brain network analysis.

脑网络域适应黎曼几何隐私保护

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