← Search

Amarda Shehu

3 accepted papers

2024

Birdie: Advancing State Space Language Modeling with Dynamic Mixtures of Training Objectives

EMNLP 2024main

Efficient state space models (SSMs), including linear recurrent neural networks and linear attention variants, have emerged as potential alternative language models to Transformers. While efficient, SSMs struggle with tasks requiring in-context retrieval, such as text copying and associative recall,…

2023

Global Convergence Analysis of Local SGD for Two-layer Neural Network without Overparameterization

NeurIPS 2023poster

Local SGD, a cornerstone algorithm in federated learning, is widely used in training deep neural networks and shown to have strong empirical performance. A theoretical understanding of such performance on nonconvex loss landscapes is currently lacking. Analysis of the global convergence of SGD is ch…

Cited by 3SourcePDFScholar
2022

Multi-objective Deep Data Generation with Correlated Property Control

NeurIPS 2022accept

Developing deep generative models has been an emerging field due to the ability to model and generate complex data for various purposes, such as image synthesis and molecular design. However, the advance of deep generative models is limited by the challenges to generate objects that possess multiple…

Cited by 12SourcePDFScholar