← Search

Chungpa Lee

6 accepted papers

2026

Fine-Tuning Without Forgetting In-Context Learning: A Theoretical Analysis of Linear Attention Models

ICML 2026poster

Transformer-based large language models exhibit in-context learning, enabling adaptation to downstream tasks via few-shot prompting with demonstrations. In practice, such models are often fine-tuned to improve zero-shot performance on downstream tasks, allowing them to solve tasks without examples a…

Cited by 0SourceScholar
2025

A Generalized Theory of Mixup for Structure-Preserving Synthetic Data

AISTATS 2025poster

Mixup is a widely adopted data augmentation technique known for enhancing the generalization of machine learning models by interpolating between data points. Despite its success and popularity, limited attention has been given to understanding the statistical properties of the synthetic data it gene…

Cited by 0SourcecodeScholar
2025

A Theoretical Framework for Preventing Class Collapse in Supervised Contrastive Learning

AISTATS 2025poster

Supervised contrastive learning (SupCL) has emerged as a prominent approach in representation learning, leveraging both supervised and self-supervised losses. However, achieving an optimal balance between these losses is challenging; failing to do so can lead to class collapse, reducing discriminati…

Cited by 0SourceScholar
2025

On the Similarities of Embeddings in Contrastive Learning

ICML 2025poster

Contrastive learning (CL) operates on a simple yet effective principle: embeddings of positive pairs are pulled together, while those of negative pairs are pushed apart. Although various forms of contrastive loss have been proposed and analyzed from different perspectives, prior works lack a compreh…