LLM-Based Feature Selection and Counterfactual Explanations Applied to Functional Connectivity Analysis in Schizophrenia

Shaolong Wei^1*Xinyan Yuan²Tiantian Chen²Yanyan He²Lingling Gu²Ying Sun²

• ¹School of Artificial Intelligence and Computer Science, Nantong University, Nantong, China
• ²School of Electronics and Information, Jiangsu Vocational College of Business, Nantong, China

ABSTRACT Schizophrenia (SZ) is a complex psychiatric disorder whose neural mechanisms are still unclear. Functional connectivity (FC) provides a unique perspective for understanding its pathology, but its high-dimensional nature poses significant challenges for feature selection and model interpretation. Traditional feature selection methods, while predictive, lack the integration of prior neuroscience knowledge, resulting in limited clinical relevance. To address this, we propose an innovative framework that combines feature selection guided by a large language model (LLM) with counterfactual explanation. This framework leverages brain disease knowledge encoded by the LLM to guide dimensionality reduction of high-dimensional FC, ensuring that selected features are both statistically significant and biologically plausible. Counterfactual explanations are then used to generate causal intervention examples, which are then translated by the LLM into intuitive explanations in natural language, providing understandable and actionable clinical insights for individual patients or physicians. We validate our approach on five real-world SZ datasets and demonstrate that it not only improves model classification performance but also provides new insights into SZ analysis.