用少量标注数据实现蚊子神经脉冲的高精度病毒感染分类
SSI-GAN: Semi-Supervised Swin-Inspired Generative Adversarial Networks for Neuronal Spike Classification
- 基于Swin结构设计生成对抗网络,结合Transformer生成器与滑窗判别器
- 仅用1%-3%标注数据即达99.93%准确率,显著降低人工标注需求
- 适用于野外场景的低资源神经信号分析,适合神经病原体检测研究
蚊子是虫媒病毒疾病的主要传播媒介。手动分类其神经脉冲模式费时且成本高。现有深度学习方法多需完全标注的脉冲数据集和高度预处理的神经信号,限制了在实际场景中的大规模应用。为应对标注数据稀缺问题,本文提出一种新型生成对抗网络——半监督Swin-inspired GAN(SSI-GAN)。该模型采用Swin-inspired的滑窗判别器与基于Transformer的生成器,用于神经脉冲序列分类,进而检测病毒嗜神经性。判别器使用多头自注意力机制的扁平窗口Transformer,可捕捉稀疏高频脉冲特征。在五次感染后时间点采集的超1500万条脉冲样本上训练,仅需1%至3%标注数据即可完成寨卡病毒、登革热病毒感染或未感染三类分类。通过贝叶斯优化框架调优超参数,并在五折蒙特卡洛交叉验证中验证鲁棒性。SSI-GAN在感染后第三天达到99.93%分类准确率,仅需1%标注数据即在所有感染阶段保持高精度。相比传统监督方法,人工标注工作量减少97%-99%,性能更优。所提出的滑窗Transformer设计全面超越基线,创下脉冲神经感染分类新纪录。
原文摘要 · Abstract (English)
Mosquitos are the main transmissive agents of arboviral diseases. Manual classification of their neuronal spike patterns is very labor-intensive and expensive. Most available deep learning solutions require fully labeled spike datasets and highly preprocessed neuronal signals. This reduces the feasibility of mass adoption in actual field scenarios. To address the scarcity of labeled data problems, we propose a new Generative Adversarial Network (GAN) architecture that we call the Semi-supervised Swin-Inspired GAN (SSI-GAN). The Swin-inspired, shifted-window discriminator, together with a transformer-based generator, is used to classify neuronal spike trains and, consequently, detect viral neurotropism. We use a multi-head self-attention model in a flat, window-based transformer discriminator that learns to capture sparser high-frequency spike features. Using just 1 to 3% labeled data, SSI-GAN was trained with more than 15 million spike samples collected at five-time post-infection and recording classification into Zika-infected, dengue-infected, or uninfected categories. Hyperparameters were optimized using the Bayesian Optuna framework, and performance for robustness was validated under fivefold Monte Carlo cross-validation. SSI-GAN reached 99.93% classification accuracy on the third day post-infection with only 3% labeled data. It maintained high accuracy across all stages of infection with just 1% supervision. This shows a 97-99% reduction in manual labeling effort relative to standard supervised approaches at the same performance level. The shifted-window transformer design proposed here beat all baselines by a wide margin and set new best marks in spike-based neuronal infection classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。