← Search

Adam X. Yang

3 accepted papers

2024

Bayesian Low-rank Adaptation for Large Language Models

ICLR 2024poster

Parameter-efficient fine-tuning (PEFT) has emerged as a new paradigm for cost-efficient fine-tuning of large language models (LLMs), with low-rank adaptation (LoRA) being a widely adopted choice. However, fine-tuned LLMs often become overconfident especially when fine-tuned on small datasets. Bayesi…

Cited by 75SourcePDFScholar
2024

Instruction Tuning With Loss Over Instructions

NeurIPS 2024poster

Instruction tuning plays a crucial role in shaping the outputs of language models (LMs) to desired styles. In this work, we propose a simple yet effective method, Instruction Modelling (IM), which trains LMs by applying a loss function to the instruction and prompt part rather than solely to the out…

2023

A theory of representation learning gives a deep generalisation of kernel methods

ICML 2023poster

The successes of modern deep machine learning methods are founded on their ability to transform inputs across multiple layers to build good high-level representations. It is therefore critical to understand this process of representation learning. However, standard theoretical approaches (formally N…

Cited by 22SourcePDFScholar