← Search

Paras Jain

5 accepted papers

2022

POET: Training Neural Networks on Tiny Devices with Integrated Rematerialization and Paging

ICML 2022spotlight

Fine-tuning models on edge devices like mobile phones would enable privacy-preserving personalization over sensitive data. However, edge training has historically been limited to relatively small models with simple architectures because training is both memory and energy intensive. We present POET,…

2021

Accelerating Quadratic Optimization with Reinforcement Learning

NeurIPS 2021poster

First-order methods for quadratic optimization such as OSQP are widely used for large-scale machine learning and embedded optimal control, where many related problems must be rapidly solved. These methods face two persistent challenges: manual hyperparameter tuning and convergence time to high-accur…

2021

Contrastive Code Representation Learning

EMNLP 2021main

Recent work learns contextual representations of source code by reconstructing tokens from their context. For downstream semantic understanding tasks like code clone detection, these representations should ideally capture program functionality. However, we show that the popular reconstruction-based…

2021

Grounded Graph Decoding improves Compositional Generalization in Question Answering

EMNLP 2021finding

Question answering models struggle to generalize to novel compositions of training patterns. Current end-to-end models learn a flat input embedding which can lose input syntax context. Prior approaches improve generalization by learning permutation invariant models, but these methods do not scale to…

2021

Representing Long-Range Context for Graph Neural Networks with Global Attention

NeurIPS 2021poster

Graph neural networks are powerful architectures for structured datasets. However, current methods struggle to represent long-range dependencies. Scaling the depth or width of GNNs is insufficient to broaden receptive fields as larger GNNs encounter optimization instabilities such as vanishing gradi…