Information Bottleneck-Guided Heterogeneous Graph Learning for Interpretable Neurodevelopmental Disorder Diagnosis

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Yueyang, Chen, Lei, Dong, Wenhao, Gong, Shengyu, Kang, Zijian, Wei, Boyang, Zeng, Weiming, Yan, Hongjie, Bian, Lingbin, Zhang, Zhiguo, Siok, Wai Ting, Wang, Nizhuan
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910159589081088
author Li, Yueyang
Chen, Lei
Dong, Wenhao
Gong, Shengyu
Kang, Zijian
Wei, Boyang
Zeng, Weiming
Yan, Hongjie
Bian, Lingbin
Zhang, Zhiguo
Siok, Wai Ting
Wang, Nizhuan
author_facet Li, Yueyang
Chen, Lei
Dong, Wenhao
Gong, Shengyu
Kang, Zijian
Wei, Boyang
Zeng, Weiming
Yan, Hongjie
Bian, Lingbin
Zhang, Zhiguo
Siok, Wai Ting
Wang, Nizhuan
contents Developing interpretable models for neurodevelopmental disorders (NDDs) diagnosis presents significant challenges in effectively encoding, decoding, and integrating multimodal neuroimaging data. While many existing machine learning approaches have shown promise in brain network analysis, they typically suffer from limited interpretability, particularly in extracting meaningful biomarkers from functional magnetic resonance imaging (fMRI) data and establishing clear relationships between imaging features and demographic characteristics. Besides, current graph neural network methodologies face limitations in capturing both local and global functional connectivity patterns while simultaneously achieving theoretically principled multimodal data fusion. To address these challenges, we propose the Interpretable Information Bottleneck Heterogeneous Graph Neural Network (I2B-HGNN), a unified framework that applies information bottleneck principles to guide both brain connectivity modeling and cross-modal feature integration. This framework comprises two complementary components. The first is the Information Bottleneck Graph Transformer (IBGraphFormer), which combines transformer-based global attention mechanisms with graph neural networks through information bottleneck-guided pooling to identify sufficient biomarkers. The second is the Information Bottleneck Heterogeneous Graph Attention Network (IB-HGAN), which employs meta-path-based heterogeneous graph learning with structural consistency constraints to achieve interpretable fusion of neuroimaging and demographic data. The experimental results demonstrate that I2B-HGNN achieves superior performance in diagnosing NDDs, exhibiting both high classification accuracy and the ability to provide interpretable biomarker identification while effectively analyzing non-imaging data.
format Preprint
id arxiv_https___arxiv_org_abs_2502_20769
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Information Bottleneck-Guided Heterogeneous Graph Learning for Interpretable Neurodevelopmental Disorder Diagnosis
Li, Yueyang
Chen, Lei
Dong, Wenhao
Gong, Shengyu
Kang, Zijian
Wei, Boyang
Zeng, Weiming
Yan, Hongjie
Bian, Lingbin
Zhang, Zhiguo
Siok, Wai Ting
Wang, Nizhuan
Computer Vision and Pattern Recognition
Developing interpretable models for neurodevelopmental disorders (NDDs) diagnosis presents significant challenges in effectively encoding, decoding, and integrating multimodal neuroimaging data. While many existing machine learning approaches have shown promise in brain network analysis, they typically suffer from limited interpretability, particularly in extracting meaningful biomarkers from functional magnetic resonance imaging (fMRI) data and establishing clear relationships between imaging features and demographic characteristics. Besides, current graph neural network methodologies face limitations in capturing both local and global functional connectivity patterns while simultaneously achieving theoretically principled multimodal data fusion. To address these challenges, we propose the Interpretable Information Bottleneck Heterogeneous Graph Neural Network (I2B-HGNN), a unified framework that applies information bottleneck principles to guide both brain connectivity modeling and cross-modal feature integration. This framework comprises two complementary components. The first is the Information Bottleneck Graph Transformer (IBGraphFormer), which combines transformer-based global attention mechanisms with graph neural networks through information bottleneck-guided pooling to identify sufficient biomarkers. The second is the Information Bottleneck Heterogeneous Graph Attention Network (IB-HGAN), which employs meta-path-based heterogeneous graph learning with structural consistency constraints to achieve interpretable fusion of neuroimaging and demographic data. The experimental results demonstrate that I2B-HGNN achieves superior performance in diagnosing NDDs, exhibiting both high classification accuracy and the ability to provide interpretable biomarker identification while effectively analyzing non-imaging data.
title Information Bottleneck-Guided Heterogeneous Graph Learning for Interpretable Neurodevelopmental Disorder Diagnosis
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.20769