← Search

ZhiqiBai

4 accepted papers

2026

DESIGNER: Design-Logic-Guided Multidisciplinary Data Synthesis for LLM Reasoning

ICLR 2026poster

Large language models (LLMs) perform strongly on many language tasks but still struggle with complex multi-step reasoning across disciplines. Existing reasoning datasets often lack disciplinary breadth, reasoning depth, and diversity, as well as guiding principles for question synthesis. We propose…

Cited by 0SourceScholar
2026

Reconstructing KV Caches with Cross-Layer Fusion for Enhanced Transformers

ICLR 2026poster

Transformer decoders have achieved strong results across tasks, but the memory required for the KV cache becomes prohibitive at long sequence lengths. Although Cross-layer KV Cache sharing (e.g., YOCO, CLA) offers a path to mitigate KV Cache bottleneck, it typically underperforms within-layer method…

Cited by 0SourcecodeScholar
2024

D-CPT Law: Domain-specific Continual Pre-Training Scaling Law for Large Language Models

NeurIPS 2024poster

Continual Pre-Training (CPT) on Large Language Models (LLMs) has been widely used to expand the model’s fundamental understanding of specific downstream domains (e.g., math and code). For the CPT on domain-specific LLMs, one important question is how to choose the optimal mixture ratio between the g…

Cited by 18SourcePDFScholar
2024

DDK: Distilling Domain Knowledge for Efficient Large Language Models

NeurIPS 2024poster

Despite the advanced intelligence abilities of large language models (LLMs) in various applications, they still face significant computational and storage demands. Knowledge Distillation (KD) has emerged as an effective strategy to improve the performance of a smaller LLM (i.e., the student model)…

Cited by 10SourcePDFScholar