← Search

Andrew Bai

7 accepted papers

2026

Self-Forcing++: Towards Minute-Scale High-Quality Video Generation

ICLR 2026poster

Diffusion models have revolutionized image and video generation, achieving unprecedented visual quality. However, their reliance on transformer architectures incurs prohibitively high computational costs, particularly when extending generation to long videos. Recent work has explored autoregressive…

Cited by 0SourcecodeScholar
2025

An Efficient Rehearsal Scheme for Catastrophic Forgetting Mitigation during Multi-stage Fine-tuning

NAACL 2025findings

Incrementally fine-tuning foundational models on new tasks or domains is now the de facto approach in NLP. A known pitfall of this approach is the catastrophic forgetting of prior knowledge that happens during fine-tuning. A common approach to alleviate such forgetting is to rehearse samples from pr…

2025

On the Loss of Context Awareness in General Instruction Fine-tuning

NeurIPS 2025poster

Pre-trained Large Language Models (LLMs) require post-training methods such as supervised fine-tuning (SFT) on instruction-response pairs to enable instruction following. However, this process can cause forgetting in capabilities learned during pre-training. In this paper, we investigate the loss of…

Cited by 0SourcecodeScholar
2025

Unlabeled Data Improves Fine-Grained Image Zero-shot Classification with Multimodal LLMs

NeurIPS 2025poster

Despite Multimodal Large Language Models (MLLMs) showing promising results on general zero-shot image classification tasks, fine-grained image classification remains challenging. It demands precise attention to subtle visual details to distinguish between visually similar subcategories—details that…

Cited by 0SourceScholar
2024

Defending LLMs against Jailbreaking Attacks via Backtranslation

ACL 2024findings

Although many large language models (LLMs) have been trained to refuse harmful requests, they are still vulnerable to jailbreaking attacks which rewrite the original prompt to conceal its harmful intent. In this paper, we propose a new method for defending LLMs against jailbreaking attacks by “backt…

2023

Concept Gradient: Concept-based Interpretation Without Linear Assumption

ICLR 2023poster

Concept-based interpretations of black-box models are often more intuitive for humans to understand. The most widely adopted approach for concept-based, gradient interpretation is Concept Activation Vector (CAV). CAV relies on learning a linear relation between some latent representation of a given…