Frequency-Domain Latent Attention Gating for Cross-Domain Token Aggregation
- URL: http://arxiv.org/abs/2606.08191v1
- Date: Sat, 06 Jun 2026 14:21:55 GMT
- Title: Frequency-Domain Latent Attention Gating for Cross-Domain Token Aggregation
- Abstract summary: Token aggregation is a common bottleneck in models that map token representations to sample-level predictions.<n>We propose FLaG, a plug-in aggregation module that transforms token representations with the real FFT.<n>We evaluate FLaG on antimicrobial peptide (AMP) activity prediction with ESM2, image classification with ResNet18, and text classification with RoBERTa.
- Score: 6.535173782861673
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: 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.
Related papers
- FLaG: Frequency-Domain Latent-attention Gated Pooling for Token Aggregation [6.298319903082888]
Token aggregation converts token-level representations into fixed-dimensional sample representations, but most pooling methods operate only in the original token space.<n>We introduce Frequency-Domain Latent-attention Gated Pooling (FLaG), a plug-in aggregation module that re-expresses encoder outputs in the domain before final pooling.<n>FLaG represents nonredundant rFFT spectrum through real and imaginary components, summarizes spectral tokens with learnable latent queries, derives a sample-conditioned gate, and reconstructs modulated token representations for downstream aggregation.
arXiv Detail & Related papers (2026-09-01T07:32:44Z) - RamanPFN: learning from Raman spectral structure with a tabular foundation model [5.156304764307903]
Raman spectroscopy enables non-destructive, label-free molecular characterization across materials science, biomedicine and process monitoring.<n> Predictive Raman datasets often contain few labelled spectra and thousands of ordered wavenumbers.<n>We present RamanPFN, a spectral representation framework that encodes dependencies before TabPFN inference.
arXiv Detail & Related papers (2026-08-03T12:37:55Z) - AMPBench-MT: A Homology-Controlled Benchmark for Antimicrobial Peptide Potency, Spectrum, and Safety Prediction [3.534208235875249]
AMP and peptide benchmarks do not jointly place AMP recognition, species-conditioned potency, spectrum, safety-facing proxy endpoints, and cross-endpoint behavior.<n> AMPBench-MT standardizes canonical peptide records and organizes them into binary recognition, species-conditioned pMIC regression, and endpoint-specific potency and safety-facing readouts.
arXiv Detail & Related papers (2026-07-28T10:01:30Z) - SagnacAssisted Enhanced OTDR for Distributed Acoustic Sensing: A Standardized Benchmark and Engineering Evaluation Framework [3.5435442578622127]
This study develops a Sagnac-assisted enhanced $$-OTDR sensing architecture and a standardized benchmark framework for engineering-oriented DAS event recognition.<n>The Sagnac interferometer provides a continuous phase response that supplements fading-prone observations in the $$-OTDR channel.<n>Experiments on a 10-km sensing fiber with six representative acoustic classes show that the dual-branch fusion model provides the most favorable trade-off among the evaluated methods.
arXiv Detail & Related papers (2026-06-04T06:29:25Z) - LimiX-2M: Mitigating Low-Rank Collapse and Attention Bottlenecks in Tabular Foundation Models [56.999481798138625]
LimiX-2M is a 2M- parameter model that outperforms larger TabPFN-v2 and TabICL baselines on widely used benchmarks.<n>We present a unified tokenize-and-route framework for strong Tabular foundation models (TFMs)<n>Results highlight value-aware tokenization and readout-aligned routing as key levers for improving the accuracy--efficiency trade-off in TFMs.
arXiv Detail & Related papers (2026-06-03T06:07:33Z) - Posterior Augmented Flow Matching [64.1559809786948]
Posterior-Augmented Flow Matching (PAFM) is a generalization of flow matching (FM)<n>We show that PAFM improves over FM by up to 3.4 FID50K across different model scales.
arXiv Detail & Related papers (2026-05-01T17:59:59Z) - SPARK-IL: Spectral Retrieval-Augmented RAG for Knowledge-driven Deepfake Detection via Incremental Learning [6.901244434385347]
SPARK-IL is a retrieval-augmented framework that combines dual-path spectral analysis with incremental learning.<n> evaluated on the UniversalFakeDetect benchmark across 19 generative models.
arXiv Detail & Related papers (2026-04-04T19:19:33Z) - StructSAM: Structure- and Spectrum-Preserving Token Merging for Segment Anything Models [57.674757786328236]
StructSAM is a resolution-preserving merge-unmerge framework tailored to Segment Anything Model (SAM)<n>We show that StructSAM reduces encoder FLOPs by 25-30% (up to 40%+ with prompt-aware merging) with minor drops in mIoU/Dice.<n>We also provide a spectral graph coarsening view showing that score-guided merging yields bounded Laplacian spectral distortion compared to random or window-restricted baselines.
arXiv Detail & Related papers (2026-03-07T18:30:58Z) - PANC: Prior-Aware Normalized Cut for Object Segmentation [0.0]
We propose a weakly supervised spectral segmentation framework that uses a minimal set of annotated visual tokens.<n>We report strong results on homogeneous, fine-grained, and texture-limited domains.<n>For multi-object benchmarks, the framework showcases explicit, user-controllable semantic segmentation.
arXiv Detail & Related papers (2026-02-06T18:07:20Z) - FASA: Frequency-aware Sparse Attention [56.26881872333624]
We propose FASA, a novel framework that achieves query-aware token eviction by dynamically predicting token importance.<n>Our key finding is that a small, identifiable subset of "dominant" FCs consistently exhibits high contextual agreement with the full attention head.<n>Across a spectrum of long-context tasks, FASA consistently outperforms all token-eviction baselines and achieves near-oracle accuracy.
arXiv Detail & Related papers (2026-02-03T06:09:06Z) - Attention-based Generative Latent Replay: A Continual Learning Approach for WSI Analysis [3.6630930118966814]
Whole slide image (WSI) classification has emerged as a powerful tool in computational pathology, but remains constrained by domain shifts.<n>We propose an Attention-based Generative Latent Replay Continual Learning framework (AGLR-CL), in a multiple instance learning (MIL) setup for domain incremental WSI classification.
arXiv Detail & Related papers (2025-05-13T12:55:46Z) - SpaRG: Sparsely Reconstructed Graphs for Generalizable fMRI Analysis [8.489318619991534]
Deep learning can help uncover patterns in resting-state functional Magnetic Resonance Imaging (rsfMRI) associated with psychiatric disorders and personal traits.
Yet the problem of interpreting deep learning findings is rarely more evident than in fMRI analyses.
We propose a simple approach to mitigate these challenges grounded on sparsification and self-supervision.
arXiv Detail & Related papers (2024-09-24T18:35:57Z) - SampleAttention: Near-Lossless Acceleration of Long Context LLM Inference with Adaptive Structured Sparse Attention [53.4441894198495]
Large language models (LLMs) now support extremely long context windows.<n>The quadratic complexity of vanilla attention results in significantly long Time-to-First-Token (TTFT) latency.<n>We propose SampleAttention, an adaptive structured and near-lossless sparse attention.
arXiv Detail & Related papers (2024-06-17T11:05:15Z) - Suppressing Mislabeled Data via Grouping and Self-Attention [60.14212694011875]
Deep networks achieve excellent results on large-scale clean data but degrade significantly when learning from noisy labels.
This paper proposes a conceptually simple yet efficient training block, termed as Attentive Feature Mixup (AFM)
It allows paying more attention to clean samples and less to mislabeled ones via sample interactions in small groups.
arXiv Detail & Related papers (2020-10-29T13:54:16Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.