arXiv:2606.08191cs.LGcs.AI2026-06

通过频域注意力门控提升跨领域特征聚合效果

Frequency-Domain Latent Attention Gating for Cross-Domain Token Aggregation

  • 将令牌表示转换至频域,用可学习查询汇总谱成分
  • 在抗菌肽预测和图像分类中显著提升性能,尤其在CIFAR-100上
  • 适合需要跨域特征融合的生物序列与多模态任务

Token聚合是将令牌表征映射到样本级预测的常见瓶颈,但多数池化方法仅在原始令牌域操作。本文提出FLaG,一种即插即用的聚合模块:通过实数快速傅里叶变换(real FFT)转换令牌表示,使用可学习的潜在查询总结谱成分,应用通道级门控,并重建增强的时域令牌用于最终池化。我们在ESM2-8M的抗菌肽(AMP)活性预测、ResNet18在CIFAR-10和CIFAR-100上的图像分类、以及RoBERTa在IMDB和GLUE上的文本分类任务上评估FLaG。结果表明,其在ESM2-8M AMP任务和CIFAR-100上表现最显著,同时在IMDB和GLUE上仍保持强文本基线竞争力。进一步通过带通剔除、门控统计、残基扰动、潜在查询读出及结构代理分层等分析发现:低频带贡献最大,高频模式更具样本特异性;门控作为共享的谱重加权阶段,交叉注意力模式具样本特异性且查询间差异微弱;高α螺旋肽在两类细菌中均表现出更强的平均谱敏感性。补充材料、源代码与数据已公开于https://www.healthinformaticslab.org/supp/ 和 https://github.com/Kewei2023/AMPCliff/tree/FLaG。

原文摘要 · Abstract (English)

Token aggregation is a common bottleneck in models that map token representations to sample-level predictions, yet most pooling methods operate only in the original token domain. We propose FLaG, a plug-in aggregation module that transforms token representations with the real FFT, summarizes spectral components with learnable latent queries, applies a channel-wise gate, and reconstructs enhanced time-domain tokens for final pooling. We evaluate FLaG on antimicrobial peptide (AMP) activity prediction with ESM2, image classification with ResNet18 on CIFAR-10 and CIFAR-100, and text classification with RoBERTa on IMDB and GLUE. FLaG achieves its clearest gains on the ESM2-8M antimicrobial peptide tasks and on CIFAR-100, while remaining competitive with strong text baselines on IMDB and GLUE. Then we probe its behavior on the AMP setting with band knockouts, gate summaries, residue perturbations, latent-query readouts, and structure-proxy stratification. We find that low-frequency bands contribute the most overall, and the remaining higher-band pattern is more sample-specific. The gate acts as a broadly shared spectral reweighting stage and the cross-attention patterns are sample-specific with mild query-wise differentiation, and higher-helix peptides exhibit stronger average spectral sensitivity in both bacteria. The supplementary materials, source code and data are released at https://www.healthinformaticslab.org/supp/ and https://github.com/Kewei2023/AMPCliff/tree/FLaG.

频域建模特征聚合跨域学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。