mlsanityData to 7 Oct 2026

How Do Decoder-Only LLMs Perceive Users? Rethinking Attention Masking for User Representation Learning

arXiv 2602.10622 ↗First posted 11 Feb 2026cs.CLcited by 3

On the map: LLM reasoning → Reasoning benchmarks and evaluation

Abstract (arXiv)

Decoder-only large language models are increasingly used as behavioral encoders for user representation learning, yet the impact of attention masking on the quality of user embeddings remains underexplored. In this work, we conduct a systematic study of causal, hybrid, and bidirectional attention masks within a unified contrastive learning framework trained on large-scale real-world Alipay data that integrates long-horizon heterogeneous user behaviors. To improve training dynamics when transitioning from causal to bidirectional attention, we propose Gradient-Guided Soft Masking, a gradient-based pre-warmup applied before a linear scheduler that gradually opens future attention during optimization. Evaluated on 9 industrial user cognition benchmarks covering prediction, preference, and marketing sensitivity tasks, our approach consistently yields more stable training and higher-quality bidirectional representations compared with causal, hybrid, and scheduler-only baselines, while remaining compatible with decoder pretraining. Overall, our findings highlight the importance of masking design and training transition in adapting decoder-only LLMs for effective user representation learning. Our code is available at https://github.com/JhCircle/Deepfind-GGSM.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Reasoning benchmarks and evaluation, one of 17 directions in LLM reasoning. Builds on 14 of its 30 references that are in the map. First posted on arXiv 11 Feb 2026.

Builds on

14 of 30 references.

Closest ancestors · 2

Same direction, newest first

  1. 2025Qwen3 Technical Report14 May
  2. 2024Rethinking Interpretability in the Era of Large Language Models30 Jan

Other references · 12

  1. 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
  2. 2024Qwen2 Technical Reportcited by 2,076
  3. 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,175
  4. 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 705
  5. 2019Passage Re-ranking with BERTcited by 629
  6. 2025Llama-Embed-Nemotron-8B: A Universal Text Embedding Model for Multilingual and Cross-Lingual Taskscited by 46
  7. 2024A Survey on Data Synthesis and Augmentation for Large Language Modelscited by 40
  8. 2025KaLM-Embedding: Superior Training Data Brings A Stronger Embedding Modelcited by 36
  9. 2025Kardia-R1: Unleashing LLMs to Reason toward Understanding and Empathy for Emotional Support via Rubric-as-Judge Reinforcement Learningcited by 22
  10. 2024Linq-Embed-Mistral Technical Reportcited by 18
  11. 2021One4all User Representation for Recommender Systems in E-commercecited by 12
  12. 2025Instruction-aware User Embedding via Synergistic Language and Representation Modelingcited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024User-LLM: Efficient LLM Contextualization with User Embeddings
  2. 2024Bitune: Leveraging Bidirectional Attention to Improve Decoder-Only LLMs
  3. 2023Can Large Language Models Change User Preference Adversarially?
  4. 2025Attention Sinks in Diffusion Language Models
  5. 2026On the Nature of Attention Sink that Shapes Decoding Strategy in Omni-LLMs
  6. 2026Preference Heads in Large Language Models: A Mechanistic Framework for Interpretable Personalization
  7. 2026PreDiff-LM: Pretrained Discrete Masked Diffusion Language Modeling with Hybrid Attention
  8. 2022Learning Large-scale Universal User Representation with Sparse Mixture of Experts