Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data.

Zhikai Jia

dblp:321/8230 · DBLP profile ↗
← Back
1ranked-venue papers
0as first author
1since 2021 · last 2025
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
1 paper
Language models and text generation · 100%

Topics — the 3 heaviest of 3, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Natural language and speech › Language models and text generation › retrieval augmentation
in-context retrieval
0.912025
Mask-Enhanced Autoregressive Prediction: Pay Less Attention to Learn More · ICML 2025
Natural language and speech › Language models and text generation
masked language modeling
0.912025
Mask-Enhanced Autoregressive Prediction: Pay Less Attention to Learn More · ICML 2025
Natural language and speech › Language models and text generation › large language model training
pre-training objectives
0.912025
Mask-Enhanced Autoregressive Prediction: Pay Less Attention to Learn More · ICML 2025

Methods — techniques the papers use, named apart from their topics

next-token prediction · 0.9masked language modeling · 0.9decoder-only transformer · 0.9
YearPublicationVenuePosition
2025 Mask-Enhanced Autoregressive Prediction: Pay Less Attention to Learn More
abstract
Large Language Models (LLMs) are discovered to suffer from accurately retrieving key information. To address this, we propose Mask-Enhanced Autoregressive Prediction (MEAP), a simple yet effective training paradigm that seamlessly integrates Masked Language Modeling (MLM) into Next-Token Prediction (NTP) to enhance the latter’s in-context retrieval capabilities. Specifically, MEAP first randomly masks a small fraction of input tokens and then directly performs the standard next-token prediction autoregressive using a decoder-only Transformer. MEAP eliminates the need for bidirectional attention or encoder-decoder architectures for MLM, incurring no additional computational overhead during pre-training or inference. Intensive experiments demonstrate that MEAP substantially outperforms NTP on key information retrieval and long-context reasoning tasks, while performing on par or better on commonsense reasoning tasks. The benefits of MEAP also extend to supervised fine-tuning, where it shows remarkable advantages in lost-in-the-middle scenarios, outperforming NTP by 11.77% percentage points. Our analysis indicates that MEAP’s effectiveness arises from its ability to promote more distinguishable attention scores by concentrating on a reduced set of non-masked tokens. This mechanism improves the model’s focus on task-relevant signals while mitigating the influence of peripheral context. These findings position MEAP as a promising training paradigm for large language models. Code has been submitted.
Xialie Zhuang, Zhikai Jia, Zhenyu Zhang 0015, Li Shen 0008, Shiwei Liu 0003
ICML2