LSTGINet: Local Attention Spatio-Temporal Graph Inference Network for Age Prediction
There is a close correlation between brain aging and age. However, traditional neural networks cannot fully capture the potential correlation between age and brain aging due to the limited receptive field. Furthermore, they are more concerned with deep spatial semantics, ignoring the fact that effec...
Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2025-03-01
|
| Series: | Algorithms |
| Subjects: | |
| Online Access: | https://www.mdpi.com/1999-4893/18/3/138 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| Summary: | There is a close correlation between brain aging and age. However, traditional neural networks cannot fully capture the potential correlation between age and brain aging due to the limited receptive field. Furthermore, they are more concerned with deep spatial semantics, ignoring the fact that effective temporal information can enrich the representation of low-level semantics. To address these limitations, a local attention spatio-temporal graph inference network (LSTGINet) was developed to explore the details of the association between age and brain aging, taking into account both spatio-temporal and temporal perspectives. First, multi-scale temporal and spatial branches are used to increase the receptive field and model the age information simultaneously, achieving the perception of static correlation. Second, these spatio-temporal feature graphs are reconstructed, and large topographies are constructed. The graph inference node aggregation and transfer functions fully capture the hidden dynamic correlation between brain aging and age. A new local attention module is embedded in the graph inference component to enrich the global context semantics, establish dependencies and interactivity between different spatio-temporal features, and balance the differences in the spatio-temporal distribution of different semantics. We use a newly designed weighted loss function to supervise the learning of the entire prediction framework to strengthen the inference process of spatio-temporal correlation. The final experimental results show that the MAE on baseline datasets such as CamCAN and NKI are 6.33 and 6.28, respectively, better than the current state-of-the-art age prediction methods, and provides a basis for assessing the state of brain aging in adults. |
|---|---|
| ISSN: | 1999-4893 |