BRIEF: BRain-Inspired network connection search with Extensive temporal feature Fusion enhances disease classification

2508.11732v1 cs.LG, cs.AI 2025-08-19
Авторы:

Xiangxiang Cui, Min Zhao, Dongmei Zhi, Shile Qi, Vince D Calhoun, Jing Sui

Резюме на русском

#### Контекст Функциональная резонансная магнитная рисунка (fMRI) широко используется для классификации различных психиатрических заболеваний, таких как шизофрения (SZ) и аутизм (ASD). Однако существующие модели сверточных нейронных сетей (CNN) испытывают проблемы в оптимизации сетевой архитектуры и эффективной функции объединения признаков. Эти ограничения приводят к неудовлетворительному классификационному результату. Вдохновленной механизмом человеческого мозга, обновляющего нейронов в процессе обучения и принятия решений, предлагается новая модель, которая способна автоматически оптимизировать структуру сети с помощью бейзлайн-решения и многослойного модуля передачи признаков по времени. #### Метод Модель BRIEF (BRain-Inspired feature Fusion) оптимизирует структуру сети с использованием усовершенствованного Q-learning и Markov Decision Process (MDP), чтобы динамически оптимизировать сеть под конкретные данные. Модель включает четыре кодировщика, которые извлекают различные типы представлений временных рядов: статические (FNC), динамические (dFNC), дисперсионный энтропийный индекс (MsDE) и потоковая информация. Для объединения этих функций использовалась модель передачи признаков с использованием Transformer, которая включает в себя стабильные и временно изменяющиеся связи, а также многомерные зависимости в разных областях мозга. Для улучшения интерпретируемости включен модуль внимательности. #### Результаты Модель BRIEF обеспечила значительные улучшения в классификации SZ и ASD по сравнению с 21 современными моделями. Для SZ достигла AUC от 91.5% до 0.6%, а для ASD от 78.4% до 0.5%. Эти результаты превышают все существующие решения, особенно в сложной области классификации динамических признаков. Это первое использование решений, основанных на имитации биологических процессов мозга, для оптимизации сети и классификации с fMRI. #### Значимость BRIEF может применяться в различных медицинских областях, включая классификацию различных психиатрических заболеваний. Она обеспечивает более точные показатели классификации, чем существующие модели. Кроме того, модель может применяться для исследования биологических признаков в связи с развитием характеристик заболеваний, что может облегчить развитие новых терапий. #### Выводы Наша модель BRIEF демонстрирует значительные улучшения в классификации психиатрических заболеваний, основываясь на моделировании биологического процесса обучения и решения. Будущие исследования будут направляться на улучшение гибкости модели, увеличение размера выборок и расширение применений к другим типам данных.

Abstract

Existing deep learning models for functional MRI-based classification have limitations in network architecture determination (relying on experience) and feature space fusion (mostly simple concatenation, lacking mutual learning). Inspired by the human brain's mechanism of updating neural connections through learning and decision-making, we proposed a novel BRain-Inspired feature Fusion (BRIEF) framework, which is able to optimize network architecture automatically by incorporating an improved neural network connection search (NCS) strategy and a Transformer-based multi-feature fusion module. Specifically, we first extracted 4 types of fMRI temporal representations, i.e., time series (TCs), static/dynamic functional connection (FNC/dFNC), and multi-scale dispersion entropy (MsDE), to construct four encoders. Within each encoder, we employed a modified Q-learning to dynamically optimize the NCS to extract high-level feature vectors, where the NCS is formulated as a Markov Decision Process. Then, all feature vectors were fused via a Transformer, leveraging both stable/time-varying connections and multi-scale dependencies across different brain regions to achieve the final classification. Additionally, an attention module was embedded to improve interpretability. The classification performance of our proposed BRIEF was compared with 21 state-of-the-art models by discriminating two mental disorders from healthy controls: schizophrenia (SZ, n=1100) and autism spectrum disorder (ASD, n=1550). BRIEF demonstrated significant improvements of 2.2% to 12.1% compared to 21 algorithms, reaching an AUC of 91.5% - 0.6% for SZ and 78.4% - 0.5% for ASD, respectively. This is the first attempt to incorporate a brain-inspired, reinforcement learning strategy to optimize fMRI-based mental disorder classification, showing significant potential for identifying precise neuroimaging biomarkers.

Ссылки и действия