← Search

Nicholas Donald Lane

18 accepted papers

2025

DEPT: Decoupled Embeddings for Pre-training Language Models

ICLR 2025oral

Language Model pre-training uses broad data mixtures to enhance performance across domains and languages. However, training on such heterogeneous text corpora requires extensive and expensive efforts. Since these data sources vary significantly in lexical, syntactic, and semantic aspects, they cause…

Cited by 1SourcePDFScholar
2025

SparsyFed: Sparse Adaptive Federated Learning

ICLR 2025poster

Sparse training is often adopted in cross-device federated learning (FL) environments where constrained devices collaboratively train a machine learning model on private data by exchanging pseudo-gradients across heterogeneous networks. Although sparse training methods can reduce communication overh…

Cited by 0SourcePDFScholar
2024

Breaking Physical and Linguistic Borders: Multilingual Federated Prompt Tuning for Low-Resource Languages

ICLR 2024poster

Pretrained large language models (LLMs) have emerged as a cornerstone in modern natural language processing, with their utility expanding to various applications and languages. However, the fine-tuning of multilingual LLMs, particularly for low-resource languages, is fraught with challenges steming…

Cited by 17SourcePDFScholar
2024

CLUES: Collaborative Private-domain High-quality Data Selection for LLMs via Training Dynamics

NeurIPS 2024poster

Recent research has highlighted the importance of data quality in scaling large language models (LLMs). However, automated data quality control faces unique challenges in collaborative settings where sharing is not allowed directly between data silos. To tackle this issue, this paper proposes a nove…

Cited by 2SourcePDFScholar
2024

Recurrent Early Exits for Federated Learning with Heterogeneous Clients

ICML 2024poster

Federated learning (FL) has enabled distributed learning of a model across multiple clients in a privacy-preserving manner. One of the main challenges of FL is to accommodate clients with varying hardware capacities; clients have differing compute and memory requirements. To tackle this challenge, r…

2024

TinyTrain: Resource-Aware Task-Adaptive Sparse Training of DNNs at the Data-Scarce Edge

ICML 2024poster

On-device training is essential for user personalisation and privacy. With the pervasiveness of IoT devices and microcontroller units (MCUs), this task becomes more challenging due to the constrained memory and compute resources, and the limited availability of labelled user data. Nonetheless, prior…

2024

Towards Neural Architecture Search through Hierarchical Generative Modeling

ICML 2024poster

Neural Architecture Search (NAS) aims to automate deep neural network design across various applications, while a good search space design is core to NAS performance. A too-narrow search space may fail to cover diverse task requirements, whereas a too-broad one can escalate computational expenses an…

Cited by 0SourcePDFScholar
2023

FedL2P: Federated Learning to Personalize

NeurIPS 2023poster

Federated learning (FL) research has made progress in developing algorithms for distributed learning of global models, as well as algorithms for local personalization of those common models to the specifics of each client’s local data distribution. However, different FL problems may require differen…

2022

BLOX: Macro Neural Architecture Search Benchmark and Algorithms

NeurIPS 2022accept

Neural architecture search (NAS) has been successfully used to design numerous high-performance neural networks. However, NAS is typically compute-intensive, so most existing approaches restrict the search to decide the operations and topological structure of a single block only, then the same block…

2022

Conditioning Sequence-to-sequence Networks with Learned Activations

ICLR 2022poster

Conditional neural networks play an important role in a number of sequence-to-sequence modeling tasks, including personalized sound enhancement (PSE), speaker dependent automatic speech recognition (ASR), and generative modeling such as text-to-speech synthesis. In conditional neural networks, the o…

Cited by 13SourcePDFScholar
2022

Do We Need Anisotropic Graph Neural Networks?

ICLR 2022poster

Common wisdom in the graph neural network (GNN) community dictates that anisotropic models---in which messages sent between nodes are a function of both the source and target node---are required to achieve state-of-the-art performance. Benchmarks to date have demonstrated that these models perform b…

2022

Prospect Pruning: Finding Trainable Weights at Initialization using Meta-Gradients

ICLR 2022poster

Pruning neural networks at initialization would enable us to find sparse models that retain the accuracy of the original network while consuming fewer computational resources for training and inference. However, current methods are insufficient to enable this optimization and lead to a large degrada…

2022

ZeroFL: Efficient On-Device Training for Federated Learning with Local Sparsity

ICLR 2022poster

When the available hardware cannot meet the memory and compute requirements to efficiently train high performing machine learning models, a compromise in either the training quality or the model complexity is needed. In Federated Learning (FL), nodes are orders of magnitude more constrained than tra…

Cited by 77SourcePDFScholar
2021

A Channel Coding Benchmark for Meta-Learning

NeurIPS 2021poster

Meta-learning provides a popular and effective family of methods for data-efficient learning of new tasks. However, several important issues in meta-learning have proven hard to study thus far. For example, performance degrades in real-world settings where meta-learners must learn from a wide and po…

Cited by 11SourcecodeScholar
2021

Degree-Quant: Quantization-Aware Training for Graph Neural Networks

ICLR 2021poster

Graph neural networks (GNNs) have demonstrated strong performance on a wide variety of tasks due to their ability to model non-uniform structured data. Despite their promise, there exists little research exploring methods to make them more efficient at inference time. In this work, we explore the vi…

Cited by 219SourcePDFScholar
2021

FjORD: Fair and Accurate Federated Learning under heterogeneous targets with Ordered Dropout

NeurIPS 2021spotlight

Federated Learning (FL) has been gaining significant traction across different ML tasks, ranging from vision to keyboard predictions. In large-scale deployments, client heterogeneity is a fact and constitutes a primary problem for fairness, training performance and accuracy. Although significant eff…

2021

NAS-Bench-ASR: Reproducible Neural Architecture Search for Speech Recognition

ICLR 2021poster

Powered by innovations in novel architecture design, noise tolerance techniques and increasing model capacity, Automatic Speech Recognition (ASR) has made giant strides in reducing word-error-rate over the past decade. ASR models are often trained with tens of thousand hours of high quality speech d…

Cited by 86SourcePDFScholar
2021

Zero-Cost Proxies for Lightweight NAS

ICLR 2021poster

Neural Architecture Search (NAS) is quickly becoming the standard methodology to design neural network models. However, NAS is typically compute-intensive because multiple models need to be evaluated before choosing the best one. To reduce the computational power and time needed, a proxy task is oft…