← Search

Devansh Arpit

13 accepted papers

2024

Retroformer: Retrospective Large Language Agents with Policy Gradient Optimization

ICLR 2024spotlight

Recent months have seen the emergence of a powerful new trend in which large language models (LLMs) are augmented to become autonomous language agents capable of performing objective oriented multi-step tasks on their own, rather than merely responding to queries from human users. Most existing lang…

2022

Ensemble of Averages: Improving Model Selection and Boosting Performance in Domain Generalization

NeurIPS 2022accept

In Domain Generalization (DG) settings, models trained independently on a given set of training domains have notoriously chaotic performance on distribution shifted test domains, and stochasticity in optimization (e.g. seed) plays a big role. This makes deep learning models unreliable in real world…

2021

Catastrophic Fisher Explosion: Early Phase Fisher Matrix Impacts Generalization

ICML 2021spotlight

The early phase of training a deep neural network has a dramatic effect on the local curvature of the loss function. For instance, using a small learning rate does not guarantee stable optimization because the optimization trajectory has a tendency to steer towards regions of the loss surface with i…

Cited by 80SourcePDFScholar
2020

The Break-Even Point on Optimization Trajectories of Deep Neural Networks

ICLR 2020spotlight

The early phase of training of deep neural networks is critical for their final performance. In this work, we study how the hyperparameters of stochastic gradient descent (SGD) used in the early phase of training affect the rest of the optimization trajectory. We argue for the existence of the "``br…

Cited by 191SourceScholar
2019

How to Initialize your Network? Robust Initialization for WeightNorm & ResNets

NeurIPS 2019poster

Residual networks (ResNet) and weight normalization play an important role in various deep learning applications. However, parameter initialization strategies have not been studied previously for weight normalized networks and, in practice, initialization methods designed for un-normalized networks…

2019

On the Spectral Bias of Neural Networks

ICML 2019oral

Neural networks are known to be a class of highly expressive functions able to fit even random input-output mappings with 100% accuracy. In this work we present properties of neural networks that complement this aspect of expressivity. By using tools from Fourier analysis, we highlight a learning bi…

2019

h-detach: Modifying the LSTM Gradient Towards Better Optimization

ICLR 2019poster

Recurrent neural networks are known for their notorious exploding and vanishing gradient problem (EVGP). This problem becomes more evident in tasks where the information needed to correctly solve them exist over long time scales, because EVGP prevents important gradient components from being back-pr…

2018

Residual Connections Encourage Iterative Inference

ICLR 2018poster

Residual networks (Resnets) have become a prominent architecture in deep learning. However, a comprehensive understanding of Resnets is still a topic of ongoing research. A recent view argues that Resnets perform iterative refinement of features. We attempt to further expose properties of this aspec…

Cited by 174SourcePDFScholar
2017

A Closer Look at Memorization in Deep Networks

ICML 2017poster

We examine the role of memorization in deep learning, drawing connections to capacity, generalization, and adversarial robustness. While deep networks are capable of memorizing noise data, our results suggest that they tend to prioritize learning simple patterns first. In our experiments, we expose…

Cited by 2324SourcePDFScholar
2016

Normalization Propagation: A Parametric Technique for Removing Internal Covariate Shift in Deep Networks

ICML 2016poster

While the authors of Batch Normalization (BN) identify and address an important problem involved in training deep networks– \textitInternal Covariate Shift– the current solution has certain drawbacks. For instance, BN depends on batch statistics for layerwise input normalization during training whic…

Cited by 154SourcePDFScholar
2016

Why Regularized Auto-Encoders learn Sparse Representation?

ICML 2016poster

Sparse distributed representation is the key to learning useful features in deep learning algorithms, because not only it is an efficient mode of data representation, but also – more importantly – it captures the generation process of most real world data. While a number of regularized auto-encoders…

Cited by 97SourcePDFScholar