← Search

Chun Cao

5 accepted papers

2026

Long-Context Attention Benchmark: From Kernel Efficiency to Distributed Context Parallelism

ICLR 2026poster

Transformer-based large language models (LLMs) have achieved remarkable success, yet their standard attention mechanism incurs quadratic computation and memory costs with respect to sequence length, posing a major bottleneck for long-context training. Prior work tackles this challenge along two dire…

Cited by 0SourcecodeScholar
2025

Loquetier: A Virtualized Multi-LoRA Framework for Unified LLM Fine-tuning and Serving

NeurIPS 2025poster

Low-Rank Adaptation (LoRA) has become a widely adopted parameter-efficient fine-tuning (PEFT) technique for adapting large language models (LLMs) to downstream tasks. While prior work has explored strategies for integrating LLM training and serving, there still remains a gap in unifying fine-tuning…

Cited by 0SourcecodeScholar
2024

Neuro-Symbolic Data Generation for Math Reasoning

NeurIPS 2024poster

A critical question about Large Language Models (LLMs) is whether their apparent deficiency in mathematical reasoning is inherent, or merely a result of insufficient exposure to high-quality mathematical data. To explore this, we developed an automated method for generating high-quality, supervised…

Cited by 7SourcePDFScholar
2023

Softened Symbol Grounding for Neuro-symbolic Systems

ICLR 2023poster

Neuro-symbolic learning generally consists of two separated worlds, i.e., neural network training and symbolic constraint solving, whose success hinges on symbol grounding, a fundamental problem in AI. This paper presents a novel, softened symbol grounding process, bridging the gap between the two…

2022

A Deep Learning Dataloader with Shared Data Preparation

NeurIPS 2022accept

Executing a family of Deep Neural Networks (DNNs) training jobs on the same or similar datasets in parallel is typical in current deep learning scenarios. It is time-consuming and resource-intensive because each job repetitively prepares (i.e., loads and preprocesses) the data independently, causing…

Cited by 9SourcePDFScholar