GeneCytNet: an interpretable deep learning framework for rheumatoid arthritis classification and in silico cytokine perturbation modeling

TL;DR

GeneCytNet is a deep learning model that improves RA diagnosis and provides insights into cytokine effects using gene data.

Contribution

Introduces an interpretable deep learning framework combining VAE and GAT for RA classification and cytokine perturbation modeling.

Findings

GeneCytNet achieved a test AUC of 0.962, outperforming baseline models in RA classification.

In silico experiments showed IL-6 has the strongest effect on RA probability, followed by TNF-α and IL-1β.

The model's robustness was confirmed with cross-validation (mean AUC = 0.957).

Abstract

Rheumatoid arthritis (RA) is a heterogeneous autoimmune disease where cytokine-driven dysregulation of gene networks poses a significant challenge for accurate diagnosis and targeted therapy. While transcriptomic data hold immense promise, most machine learning models lack the interpretability to decipher the underlying biological mechanisms, particularly the specific roles of key cytokines.

We developed GeneCytNet, a novel deep learning framework that integrates a Variational Autoencoder (VAE) for nonlinear feature compression with a Graph Attention Network (GAT) to model gene-gene interactions. The model was developed on a synthetic cohort of 240 RA and 120 healthy control samples, with an independent holdout cohort of 100 RA and 50 controls, each with 15,000 gene features, designed as a robust proof-of-concept. Performance was benchmarked against classical models, and generalizability was assessed via cross-validation and the independent holdout. Crucially, we introduced in silico cytokine perturbation experiments to simulate the effect of modulating IL-6, TNF-α, and IL-1β responsive gene modules on RA risk prediction.

GeneCytNet achieved superior classification performance, with a test AUC of 0.962 ± 0.005, accuracy of 0.914 ± 0.007, and an F1-score of 0.915 ± 0.006, outperforming all baseline models. Cross-validation confirmed robustness (mean AUC = 0.957 ± 0.006). The perturbation experiments provided mechanistically interpretable insights, revealing that the IL-6–responsive module had the most significant effect on RA probability (+0.12 ± 0.03), followed by TNF-α (+0.08 ± 0.02) and IL-1β (+0.06 ± 0.02). This hierarchy of cytokine effect sizes aligns with established clinical evidence.

GeneCytNet demonstrates that advanced, interpretable deep learning can simultaneously achieve high diagnostic accuracy and generate testable biological hypotheses. By functioning as a virtual patient simulator, our framework bridges the gap between prediction and mechanism, offering a powerful tool for precision diagnostics, biomarker discovery, and the design of cytokine-targeted therapies in RA and other complex diseases.

Figures