← Search

Zi Gong

2 accepted papers

2025

Rodimus*: Breaking the Accuracy-Efficiency Trade-Off with Efficient Attentions

ICLR 2025poster

Recent advancements in Transformer-based large language models (LLMs) have set new standards in natural language processing. However, the classical softmax attention incurs significant computational costs, leading to a $O(T)$ complexity for per-token generation, where $T$ represents the context leng…

2024

CoBa: Convergence Balancer for Multitask Finetuning of Large Language Models

EMNLP 2024main

Multi-task learning (MTL) benefits the fine-tuning of large language models (LLMs) by providing a single model with improved performance and generalization ability across tasks, presenting a resource-efficient alternative to developing separate models for each task. Yet, existing MTL strategies for…