Saved in:
| Main Authors: | , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2506.15708 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866915350528917504 |
|---|---|
| author | Febrinanto, Falih Gozi Simango, Adonia Xu, Chengpei Zhou, Jingjing Ma, Jiangang Tyagi, Sonika Xia, Feng |
| author_facet | Febrinanto, Falih Gozi Simango, Adonia Xu, Chengpei Zhou, Jingjing Ma, Jiangang Tyagi, Sonika Xia, Feng |
| contents | Graph neural networks (GNNs) have been developed to model the relationship between regions of interest (ROIs) in brains and have shown significant improvement in detecting brain diseases. However, most of these frameworks do not consider the intrinsic relationship of causality factor between brain ROIs, which is arguably more essential to observe cause and effect interaction between signals rather than typical correlation values. We propose a novel framework called CGB (Causal Graphs for Brains) for brain disease classification/detection, which models refined brain networks based on the causal discovery method, transfer entropy, and geometric curvature strategy. CGB unveils causal relationships between ROIs that bring vital information to enhance brain disease classification performance. Furthermore, CGB also performs a graph rewiring through a geometric curvature strategy to refine the generated causal graph to become more expressive and reduce potential information bottlenecks when GNNs model it. Our extensive experiments show that CGB outperforms state-of-the-art methods in classification tasks on brain disease datasets, as measured by average F1 scores. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2506_15708 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Refined Causal Graph Structure Learning via Curvature for Brain Disease Classification Febrinanto, Falih Gozi Simango, Adonia Xu, Chengpei Zhou, Jingjing Ma, Jiangang Tyagi, Sonika Xia, Feng Machine Learning Artificial Intelligence Graph neural networks (GNNs) have been developed to model the relationship between regions of interest (ROIs) in brains and have shown significant improvement in detecting brain diseases. However, most of these frameworks do not consider the intrinsic relationship of causality factor between brain ROIs, which is arguably more essential to observe cause and effect interaction between signals rather than typical correlation values. We propose a novel framework called CGB (Causal Graphs for Brains) for brain disease classification/detection, which models refined brain networks based on the causal discovery method, transfer entropy, and geometric curvature strategy. CGB unveils causal relationships between ROIs that bring vital information to enhance brain disease classification performance. Furthermore, CGB also performs a graph rewiring through a geometric curvature strategy to refine the generated causal graph to become more expressive and reduce potential information bottlenecks when GNNs model it. Our extensive experiments show that CGB outperforms state-of-the-art methods in classification tasks on brain disease datasets, as measured by average F1 scores. |
| title | Refined Causal Graph Structure Learning via Curvature for Brain Disease Classification |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2506.15708 |