Brain community detection is an efficient method to represent the communities of brain networks. However, time-variable functions of the brain and the intricate brain community structure impose a great challenge on it. In this paper, a time-sequential graph adversarial learning (TGAL) framework is proposed to detect brain communities and characterize the structure of communities from brain networks. In the framework, a novel time-sequential graph neural network is designed as an encoder to extract efficient graph representations by spatio-temporal attention mechanism. Since it is difficult to capture the community structure, the measurable modularity loss is used to optimize by maximizing the modularity of the community. In addition, the framework employs an adversarial scheme to guide the learning of representation. The effectiveness of our model is shown through experiments on the real-world brain network datasets, and the great performance of brain community detection demonstrates the advantage of the proposed framework.
Citation: Changwei Gong, Bing Xue, Changhong Jing, Chun-Hui He, Guo-Cheng Wu, Baiying Lei, Shuqiang Wang. Time-sequential graph adversarial learning for brain modularity community detection[J]. Mathematical Biosciences and Engineering, 2022, 19(12): 13276-13293. doi: 10.3934/mbe.2022621
Brain community detection is an efficient method to represent the communities of brain networks. However, time-variable functions of the brain and the intricate brain community structure impose a great challenge on it. In this paper, a time-sequential graph adversarial learning (TGAL) framework is proposed to detect brain communities and characterize the structure of communities from brain networks. In the framework, a novel time-sequential graph neural network is designed as an encoder to extract efficient graph representations by spatio-temporal attention mechanism. Since it is difficult to capture the community structure, the measurable modularity loss is used to optimize by maximizing the modularity of the community. In addition, the framework employs an adversarial scheme to guide the learning of representation. The effectiveness of our model is shown through experiments on the real-world brain network datasets, and the great performance of brain community detection demonstrates the advantage of the proposed framework.
[1] | V. R. Steele, J. M. Maurer, M. R. Arbabshirani, E. D. Claus, B. C. Fink, V. Rao, et al., Machine learning of functional magnetic resonance imaging network connectivity predicts substance abuse treatment completion, Biol. Psychiatry Cognit. Neurosci. Neuroimaging, 3 (2018), 141–149. https://doi.org/10.1016/j.bpsc.2017.07.003 doi: 10.1016/j.bpsc.2017.07.003 |
[2] | R. F. etzel, J. D. Medaglia, D. S. Bassett, Diversity of meso-scale architecture in human and non-human connectomes, Nat. Comm., 9 (2018), 346. https://doi.org/10.1038/s41467-017-02681-z doi: 10.1038/s41467-017-02681-z |
[3] | M. Rubinov, O. Sporns, Weight-conserving characterization of complex functional brain networks, Neuroimage, 56 (2011), 2068–2079. https://doi.org/10.1016/j.neuroimage.2011.03.069 doi: 10.1016/j.neuroimage.2011.03.069 |
[4] | R. F. Betzel, D. S. Bassett, Multi-scale brain networks, Neuroimage, 160 (2017), 73–83. https://doi.org/10.1016/j.neuroimage.2016.11.006 |
[5] | M. Rubinov, O. Sporns, Complex network measures of brain connectivity: Uses and interpretations, Neuroimage, 52 (2010), 1059–1069. https://doi.org/10.1016/j.neuroimage.2009.10.003 doi: 10.1016/j.neuroimage.2009.10.003 |
[6] | J. Pan, B. Lei, Y. Shen, Y. Liu, Z. Feng, S. Wang, Characterization multimodal connectivity of brain network by hypergraph GAN for Alzheimer's disease analysis, in Chinese Conference on Pattern Recognition and Computer Vision (PRCV), (2021), 467–478. https://doi.org/10.1007/978-3-030-88010-1_39 |
[7] | S. Hu, J. Yuan, S. Wang, Cross-modality synthesis from MRI to PET using adversarial U-net with different normalization, in 2019 International Conference on Medical Imaging Physics and Engineering (ICMIPE), (2019), 1–5. https://doi.org/10.1109/ICMIPE47306.2019.9098219 |
[8] | S. Hu, Y. Shen, S. Wang, B. Lei, Brain MR to PET synthesis via bidirectional generative adversarial network, in Medical Image Computing and Computer Assisted Intervention–-MICCAI 2020: 23rd International Conference, 10 (2020), 698–707. https://doi.org/10.1007/978-3-030-59713-9_67 |
[9] | O. Sporns, Contributions and challenges for network models in cognitive neuroscience, Nat. Neurosci., 17 (2014), 652–660. https://doi.org/10.1038/nn.3690 doi: 10.1038/nn.3690 |
[10] | M. E. Newman, Communities, modules and large-scale structure in networks, Nat. Phys., 8 (2012), 25–31. https://doi.org/10.1038/nphys2162 doi: 10.1038/nphys2162 |
[11] | O. Sporns, R. F. Betzel, Modular brain networks, Ann. Rev. Psychol., 67 (2016), 613–640. https://doi.org/10.1146/annurev-psych-122414-033634 |
[12] | M. A. Bertolero, B. T. T. Yeo, M. D'Esposito, The modular and integrative functional architecture of the human brain, Proc. Nat. Acad. Sci. U. S. A., 112 (2015), E6798–E6807. https://doi.org/10.1073/pnas.1510619112 doi: 10.1073/pnas.1510619112 |
[13] | M. P. van den Heuvel, O. Sporns, Network hubs in the human brain, Trends Cogn. Sci., 17 (2013), 683–696. https://doi.org/10.1016/j.tics.2013.09.012 doi: 10.1016/j.tics.2013.09.012 |
[14] | S. Fortunato, Community detection in graphs, Phys. Rep., 486 (2010), 75–174. https://doi.org/10.1016/j.physrep.2009.11.002 doi: 10.1016/j.physrep.2009.11.002 |
[15] | D. S. Bassett, O. Sporns, Network neuroscience, Nat. Neurosci., 20 (2017), 353–364. https://doi.org/10.1038/nn.4502 |
[16] | M. Girvan, M. E. J. Newman, Community structure in social and biological networks, Proc. Nat. Acad. Sci. U. S. A., 99 (2002), 7821–7826. https://doi.org/10.1073/pnas.122653799 doi: 10.1073/pnas.122653799 |
[17] | J. Pan, B. Lei, S. Wang, B. Wang, Y. Liu, Y. Shen, DecGAN: Decoupling generative adversarial network detecting abnormal neural circuits for Alzheimer's disease, preprint, arXiv: 2110.05712. |
[18] | A. Avena-Koenigsberger, B. Misic, O. Sporns, Communication dynamics in complex brain networks, Nat. Rev. Neurosci., 19 (2018), 17–33. https://doi.org/10.1038/nrn.2017.149 doi: 10.1038/nrn.2017.149 |
[19] | S. Hu, W. Yu, Z. Chen, S. Wang, Medical image reconstruction using generative adversarial network for Alzheimer disease assessment with class-imbalance problem, in 2020 IEEE 6th International Conference on Computer and Communications (ICCC), (2020), 1323–1327. https://doi.org/10.1109/ICCC51575.2020.9344912 |
[20] | S. Wang, X. Li, J. Cui, H. X. Li, K. D. Luk, Y. Hu, Prediction of myelopathic level in cervical spondylotic myelopathy using diffusion tensor imaging, J. Magn. Reson. Imaging, 41 (2015), 1682–1688. https://doi.org/10.1002/jmri.24709 doi: 10.1002/jmri.24709 |
[21] | B. Lei, E. Liang, M. Yang, P. Yang, F. Zhou, E. L. Tan, et al., Predicting clinical scores for Alzheimer's disease based on joint and deep learning, Exp. Syst. Appl., 187 (2022), 115966. https://doi.org/10.1016/j.eswa.2021.115966 doi: 10.1016/j.eswa.2021.115966 |
[22] | A. K. F. Lui, Y. H. Chan, M. F. Leung, Modelling of destinations for data-driven pedestrian trajectory prediction in public buildings, in 2021 IEEE International Conference on Big Data (Big Data), (2021), 1709–1717. https://doi.org/10.1109/BigData52589.2021.9671813 |
[23] | S. Hu, B. Lei, S. Wang, Y. Wang, Z. Feng, Y. Shen, Bidirectional mapping generative adversarial networks for brain MR to PET synthesis, IEEE Trans. Med. Imaging, 41 (2021), 145–157. https://doi.org/10.1109/TMI.2021.3107013 doi: 10.1109/TMI.2021.3107013 |
[24] | L. Yang, X. Cao, D. He, C. Wang, X. Wang, W. Zhang, Modularity based community detection with deep learning, in International Joint Conferences on Artificial Intelligence Organization, (2016), 2252–2258. https://doi.org/10.5555/3060832.3060936 |
[25] | S. Wang, X. Wang, Y. Shen, B. He, X. Zhao, W. Cheung, et al., An ensemble-based densely-connected deep learning system for assessment of skeletal maturity, IEEE Trans. Syst. Man Cybern. Syst., 52 (2020), 426–437. https://doi.org/10.1109/TSMC.2020.2997852 doi: 10.1109/TSMC.2020.2997852 |
[26] | G. Sperlí, A deep learning based community detection approach, in Proceedings of the 34th ACM/SIGAPP Symposium on Applied Computing, (2019), 1107–1110. https://doi.org/10.1145/3297280.3297574 |
[27] | S. Wang, X. Wang, Y. Hu, Y. Shen, Z. Yang, M. Gan, et al., Diabetic retinopathy diagnosis using multichannel generative adversarial network with semisupervision, IEEE Trans. Autom. Sci. Eng., 18 (2021), 574–585. https://doi.org/10.1109/TASE.2020.2981637 doi: 10.1109/TASE.2020.2981637 |
[28] | W. Yu, B. Lei, M. K. Ng, A. C. Cheung, Y. Shen, S. Wang, Tensorizing GAN with high-order pooling for Alzheimer's disease assessment, IEEE Trans. Neural Networks Learn. Syst., 2021 (2021), 1–15. https://doi.org/10.1109/TNNLS.2021.3063516 doi: 10.1109/TNNLS.2021.3063516 |
[29] | B. Lei, Z. Xia, J. Feng, X. Jiang, Z. Ge, Y. Xu, et al., Skin lesion segmentation via generative adversarial networks with dual discriminators, Med. Image Anal., 64 (2020), 101716. https://doi.org/10.1016/j.media.2020.101716 doi: 10.1016/j.media.2020.101716 |
[30] | Z. Chen, L. Li, J. Bruna, Supervised community detection with line graph neural networks, preprint, arXiv: 1705.08415. https://doi.org/10.48550/arXiv.1705.08415 |
[31] | G. Ian, P. A. Jean, M. Mehdi, Generative adversarial nets, Adv. Neural Inf. Process. Syst., 27 (2014), 1–9. |
[32] | L. Mo, S. Wang, A variational approach to nonlinear two-point boundary value problems, Nonlinear Anal. Theory Methods Appl., 71 (2009), e834–e838. https://doi.org/10.1016/j.na.2008.12.006 doi: 10.1016/j.na.2008.12.006 |
[33] | B. Jake, Notes on convolutional neural networks, Neural Nets, 2006 (2006), 1–8. |
[34] | T. N. Kipf, M. Welling, Semi-supervised classification with graph convolutional networks, preprint, arXiv: 1609.02907. |
[35] | W. Yu, B. Lei, S. Wang, Y. Liu, Z. Feng, Y. Hu, et al., Morphological feature visualization of Alzheimer's disease via multidirectional perception GAN, IEEE Trans. Neural Networks Learn. Syst., (2022), 1–15. https://doi.org/10.1109/TNNLS.2021.3118369 |
[36] | S. You, Y. Liu, B. Lei, S. Wang, Fine perceptive GANs for brain MR image super-resolution in wavelet domain, IEEE Trans. Neural Networks Learn. Syst., (2022), 1–13. https://doi.org/10.1109/TNNLS.2022.3153088 |
[37] | E. Abbe, Community detection and stochastic block models: Recent developments, J. Mach. Learn. Res., 18 (2018), 6446–6531. |
[38] | F. Liu, S. Xue, J. Wu, C. Zhou, W. Hu, C. Paris, et al., Deep learning for community detection: Progress, challenges and opportunities, in International Joint Conferences on Artificial Intelligence Organization, (2020), 4981–4987. https://doi.org/10.24963/ijcai.2020/693 |
[39] | P. Veličković, G. Cucurull, A. Casanova, A. Romero, P. Lio, Y. Bengio, Graph attention networks, preprint, arXiv: 1710.10903. |
[40] | A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez, et al., Attention is all You need, in Proceedings of the 31st International Conference on Neural Information Processing Systems, (2017), 6000–6010. https://doi.org/10.5555/3295222.3295349 |
[41] | Z. Tao, H. Liu, J. Li, Z. Wang, Y. Fu, Adversarial graph embedding for ensemble clustering, in International Joint Conferences on Artificial Intelligence Organization, (2019), 3562–3568. https://doi.org/10.24963/ijcai.2019/494 |
[42] | I. Lobov, S. Ivanov, Unsupervised community detection with modularity-based attention model, preprint, arXiv: 1905.10350. |
[43] | W. Kool, H. van Hoof, M. Welling, Attention, learn to solve routing problems, in International Conference on Learning Representations, 2019. https://doi.org/10.48550/arXiv.1803.08475 |
[44] | T. Kipf, M. Welling, Variational graph auto-encoders, preprint, arXiv: 1611.07308. |
[45] | S. Pan, R. Hu, G. Long, J. Jiang, L. Yao, C. Zhang, Adversarially regularized graph autoencoder for graph embedding, in Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence, (2018), 2609–2615. https://doi.org/10.24963/ijcai.2018/362 |
[46] | P. Veličković, W. Fedus, W. L. Hamilton, P. Lio, Y. Bengio, R. D. Hjelm, Deep graph infomax, in 7th International Conference on Learning Representations, ICLR 2019, 2018. https://doi.org/10.17863/CAM.40744 |