Han Xiangmin, Xue Rundong, Feng Jingxi, Feng Yifan, Du Shaoyi, Shi Jun, Gao Yue
IEEE Trans Neural Netw Learn Syst. 2025 Apr 17;PP. doi: 10.1109/TNNLS.2025.3554755.
The goal of the hypergraph foundation model (HGFM) is to learn an encoder based on the hypergraph computational paradigm through self-supervised pretraining on high-order correlation structures, enabling the encoder to rapidly adapt to various downstream tasks in scenarios, where no labeled data or only a small amount of labeled data are available. The initial exploratory work has been applied to brain disease diagnosis tasks. However, existing methods primarily rely on graph-based approaches to learn low-order correlation patterns between brain regions in brain networks, neglecting the modeling and learning of complex correlations between different brain diseases and patients. This article proposes an HGFM for brain disease diagnosis, which conducts multidimensional pretraining tasks to explore latent cross-dimensional high-order correlation patterns on various brain disease datasets. HGFM is a high-order correlation-driven foundation model for brain disease diagnosis and effectively improves prediction performance. Specifically, HGFM first performs brain functional network link prediction tasks on individual brain networks and group interaction network link prediction tasks on group brain networks, constructing an HGFM for brain disease diagnosis. In downstream tasks, it achieves predictions for different brain disease diagnosis tasks through few-shot learning fine-tuning methods. The proposed method is evaluated on functional magnetic resonance imaging (fMRI) data from 4409 patients across four brain diseases. Results show that it outperforms existing state-of-the-art methods in all brain disease diagnosis tasks, demonstrating its potential value in clinical applications.
超图基础模型(HGFM)的目标是通过对高阶相关结构进行自监督预训练,学习基于超图计算范式的编码器,使该编码器能够在没有标记数据或只有少量标记数据的场景中快速适应各种下游任务。初步的探索性工作已应用于脑部疾病诊断任务。然而,现有方法主要依赖基于图的方法来学习脑网络中脑区之间的低阶相关模式,而忽略了不同脑部疾病和患者之间复杂相关性的建模和学习。本文提出了一种用于脑部疾病诊断的HGFM,它在各种脑部疾病数据集上进行多维预训练任务,以探索潜在的跨维度高阶相关模式。HGFM是一种用于脑部疾病诊断的高阶相关驱动基础模型,有效提高了预测性能。具体而言,HGFM首先在个体脑网络上执行脑功能网络链接预测任务,在群体脑网络上执行群体交互网络链接预测任务,构建用于脑部疾病诊断的HGFM。在下游任务中,它通过少样本学习微调方法实现对不同脑部疾病诊断任务的预测。该方法在来自4409名患有四种脑部疾病患者的功能磁共振成像(fMRI)数据上进行了评估。结果表明,在所有脑部疾病诊断任务中,它均优于现有的最先进方法,证明了其在临床应用中的潜在价值。