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.
Ссылки и действия
Дополнительные ресурсы: