works on

From the 1 of 5 linked papers with an AI index.

collaborators

5 papers

cs.CL2026

Understanding Is Done Early: A Depth Division of Labor in Large Language Models and Its Use for Unbounded-Context Memory

Hanzuo Liu, Xuan Qi, Chunyu Liu +6

The paper proposes CoMem, a method that stores intermediate transformer layer states as memory to enable efficient long‑context retrieval, showing that using lower‑mid layers for c…

cs.CL2026

VoidPadding: Let [VOID] Handle Padding in Masked Diffusion Language Models so that [EOS] Can Focus on Semantic Termination

Chunyu Liu, Zhengyang Fan, Kaisen Yang +1

MDLMs generate text by denoising a preallocated masked response canvas, making response-length modeling central to instruction tuning. Existing MDLMs often inherit the autoregressi…

cs.LG2026

Explore-Execute Chain: Towards an Efficient Structured Reasoning Paradigm

Kaisen Yang, Tinghe Zhang, Rushi Shah +4

Many LLMs plan before they act, yet planning and execution are often still entangled in one long generation trace, enforced only through prompts, or split across separate component…

cs.CL2026

Improving Sampling for Masked Diffusion Models via Information Gain

Kaisen Yang, Jayden Teoh, Kaicheng Yang +2

Masked Diffusion Models (MDMs) enable flexible decoding orders, yet existing samplers remain largely greedy, selecting locally certain tokens without accounting for their downstrea…

cs.AI2025

Attention Schema-based Attention Control (ASAC): A Cognitive-Inspired Approach for Attention Management in Transformers

Krati Saxena, Federico Jurado Ruiz, Guido Manzi +2

Attention mechanisms have become integral in AI, significantly enhancing model performance and scalability by drawing inspiration from human cognition. Concurrently, the Attention…