UniGeM: Unifying Data Selection and Mixing via Geometric Exploration and Mining

Changhao Wang, Yunfeiyu, Xinhao Yao, Jiaolong Yang, Lu Yu, Junpeng Fang, Chaobo Li, Riccardo Cantoro, Qing Cui, Jun Zhou


Abstract
The scaling of Large Language Models (LLMs) is increasingly limited by data quality. Most methods handle data mixing and sample selection separately, which can break the structure in code corpora. We introduce UniGeM, a framework that unifies mixing and selection by treating data curation as a manifold approximation problem without training proxy models or relying on external reference datasets. UniGeM operates hierarchically: Macro-Exploration learns mixing weights with stability-based clustering; Micro-Mining filters high-quality instances by their geometric distribution to ensure logical consistency. Validated by training 8B and 16B MoE models on 100B tokens, UniGeM achieves 2.0 × data efficiency over a random baseline and further improves overall performance compared to SOTA methods in reasoning-heavy evaluations and multilingual generalization.
Anthology ID:
2026.findings-acl.1037
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
20696–20714
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1037/
DOI:
Bibkey:
Cite (ACL):
Changhao Wang, Yunfeiyu, Xinhao Yao, Jiaolong Yang, Lu Yu, Junpeng Fang, Chaobo Li, Riccardo Cantoro, Qing Cui, and Jun Zhou. 2026. UniGeM: Unifying Data Selection and Mixing via Geometric Exploration and Mining. In Findings of the Association for Computational Linguistics: ACL 2026, pages 20696–20714, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
UniGeM: Unifying Data Selection and Mixing via Geometric Exploration and Mining (Wang et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1037.pdf
Checklist:
 2026.findings-acl.1037.checklist.pdf