← Search

Liam Collins

8 accepted papers

2025

Provable Meta-Learning with Low-Rank Adaptations

NeurIPS 2025poster

The power of foundation models (FMs) lies in their capacity to learn highly expressive representations that can be adapted to a broad spectrum of tasks. However, these pretrained models require additional training stages to become effective for downstream applications. In the multi-task setting, pri…

Cited by 0SourceScholar
2024

In-Context Learning with Transformers: Softmax Attention Adapts to Function Lipschitzness

NeurIPS 2024spotlight

A striking property of transformers is their ability to perform in-context learning (ICL), a machine learning framework in which the learner is presented with a novel context during inference implicitly through some data, and tasked with making a prediction in that context. As such, that learner mus…

Cited by 20SourcePDFScholar
2024

Provable Multi-Task Representation Learning by Two-Layer ReLU Neural Networks

ICML 2024oral

An increasingly popular machine learning paradigm is to pretrain a neural network (NN) on many tasks offline, then adapt it to downstream tasks, often by re-training only the last linear layer of the network. This approach yields strong downstream performance in a variety of contexts, demonstrating…

Cited by 12SourcePDFScholar
2023

Meta-Learning for Image-Guided Millimeter-Wave Beam Selection in Unseen Environments

ICASSP 2023accepted

The use of alternate modalities, like images, for fast beamforming in the millimeter wave (mmWave)-band is being proposed to ensure high bandwidth connectivity in vehicular scenarios typically seen in the context of autonomous cars. Considering the dynamic deployment conditions, a car may encounter…

Cited by 0SourceScholar
2022

FedAvg with Fine Tuning: Local Updates Lead to Representation Learning

NeurIPS 2022accept

The Federated Averaging (FedAvg) algorithm, which consists of alternating between a few local stochastic gradient updates at client nodes, followed by a model averaging update at the server, is perhaps the most commonly used method in Federated Learning. Notwithstanding its simplicity, several empir…

Cited by 105SourcePDFScholar
2021

Exploiting Shared Representations for Personalized Federated Learning

ICML 2021spotlight

Deep neural networks have shown the ability to extract universal feature representations from data such as images and text that have been useful for a variety of learning tasks. However, the fruits of representation learning have yet to be fully-realized in federated settings. Although data in feder…

Cited by 966SourcePDFScholar