用转移矩阵损失提升注意力模型对长序列的依赖捕捉能力。
Context-Aware Regularization with Markovian Integration for Attention-Based Nucleotide Analysis
- 引入转移矩阵损失,让模型学习序列间真实转移规律。
- 在40个任务中33项准确率提升,最长序列任务性能超前7%。
- 适合基因组功能预测、调控元件识别等长序列分析场景。
Transformer已革新核苷酸序列分析,但长程依赖捕捉仍具挑战。现有自回归模型常表现出马尔可夫行为,依赖固定长度上下文窗口进行下一个词预测。标准自注意力机制因二次复杂度对长序列计算效率低,且未显式强制全局转移一致性。本文提出CARMANIA(基于注意力的核苷酸分析上下文感知正则化与马尔可夫整合框架),通过在下一词预测基础上加入转移矩阵(TM)损失,使预测转移与输入序列经验性n-gram统计对齐,从而增强对局部上下文之外高阶依赖的捕捉能力。该方法使CARMANIA学习到反映进化约束与功能组织的物种特异性序列结构。我们在多种基因组任务中评估:调控元件预测、功能基因分类、分类学推断、抗微生物耐药性检测及生物合成基因簇分类。CARMANIA在长序列任务上优于此前最佳模型至少7%,在短序列上达到当前最优水平(40项任务中20项超越先前结果,且运行速度约快2.5倍),尤其在增强子与管家基因分类任务上表现突出,增强子预测的马修斯相关系数(MCC)绝对提升最高达34%。TM损失在40项任务中的33项提升准确率,尤其在依赖局部基序或调控模式的任务中效果显著。
原文摘要 · Abstract (English)
Transformers have revolutionized nucleotide sequence analysis, yet capturing long-range dependencies remains challenging. Recent studies show that autoregressive transformers often exhibit Markovian behavior by relying on fixed-length context windows for next-token prediction. However, standard self-attention mechanisms are computationally inefficient for long sequences due to their quadratic complexity and do not explicitly enforce global transition consistency. We introduce CARMANIA (Context-Aware Regularization with Markovian Integration for Attention-Based Nucleotide Analysis), a self-supervised pretraining framework that augments next-token (NT) prediction with a transition-matrix (TM) loss. The TM loss aligns predicted token transitions with empirically derived n-gram statistics from each input sequence, encouraging the model to capture higher-order dependencies beyond local context. This integration enables CARMANIA to learn organism-specific sequence structures that reflect both evolutionary constraints and functional organization. We evaluate CARMANIA across diverse genomic tasks, including regulatory element prediction, functional gene classification, taxonomic inference, antimicrobial resistance detection, and biosynthetic gene cluster classification. CARMANIA outperforms the previous best long-context model by at least 7 percent, matches state-of-the-art on shorter sequences (exceeding prior results on 20 out of 40 tasks while running approximately 2.5 times faster), and shows particularly strong improvements on enhancer and housekeeping gene classification tasks, including up to a 34 percent absolute gain in Matthews correlation coefficient (MCC) for enhancer prediction. The TM loss boosts accuracy in 33 of 40 tasks, especially where local motifs or regulatory patterns drive prediction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。