Pinned Tweet
🚨 NEW PRE-TRAINING DATASET 🚨
"CodeAlchemy" - a pipeline for turning messy code into 850B+ tokens of rich synthetic data. One of the largest synthetic code corpora released to date🔥
⛁ huggingface.co/datasets/open…
📝 arxiv.org/abs/2606.10087
w/ Aditya Prasad, Rameswar Panda
🧵👇
(7/n) 🚨 [NEW BENCHMARKS] 🚨
We're also releasing 2 new benchmarks to expose gaps that HumanEval/MBPP don't measure:
📋 DevEval - 1488 realistic developer tasks across 12 languages (debugging, porting, refactoring...)
🔍 TraceEval - 1050 trace-prediction tasks across 14 langs
(8/n) If you're training a LLM, we'd LOVE to see CodeAlchemy in your data mix!🤗
Try it, break it and TELL US what works! 🛠️🧐
📝 Paper: arxiv.org/abs/2606.10087
⛁ Data: huggingface.co/datasets/open…
Ankit Gupta retweeted
🚀 I'm delighted to share an **open postdoc position** joint between our group at @MIT_CSAIL and @MITIBMLab, mentored by me, Ankit Gupta, @LChoshen, and friends; the start date can be as early as fall 2026.
**ONLY for MIT PhD students**: Summer 2026 internship at IBM Research Cambridge. You'll have ample compute, flexibility to work on projects aligned with agentic AI, preferably coding agents, with the aim for a publication at top-tier venues. Please send your CV to [email protected].
Ankit Gupta retweeted
We're looking for 2 interns for Summer 2026 at the MIT-IBM Watson AI Lab Foundation Models Team.
Work on RL environments, enterprise benchmarks, model architecture, efficient training and finetuning, and more!
Apply here: forms.gle/H6dNSywXCjDDyBsq7
On behalf of my coauthors @AmosaurusRex @JonathanBerant, I want to express our immense gratitude to the #ICLR2024 awards committee & thank all the reviewers for their support. We will try our best to continue participating in & contributing to the research conversation.
Honestly cannot believe that our work got the BEST PAPER award @iclr_conf !!!
This was an amazing experience with my collaborators @JonathanBerant @ankgup2 , looking forward to share with everyone at the conference.
Reach out if you want to chat!
Ankit Gupta retweeted
Happy that our work on "never train from scratch" was selected for an outstanding paper and really proud of @AmosaurusRex and @ankgup2 for their effort!
Announcing the #ICLR2024 Outstanding Paper Awards:
blog.iclr.cc/2024/05/06/iclr…
Shoutout to the awards committee: @eunsolc, @katjahofmann, @liu_mingyu, @nanjiang_cs, @guennemann, @optiML, @tkipf, @CevherLIONS
Congratulations to @AmosaurusRex for his work on improving the evaluation of long-range models via self-pretraining, getting accepted to #ICLR2024 for oral presentation (top 1.2% of all submissions)!! 👏👏
w/ @JonathanBerant
Excited to share my work with @JonathanBerant @ankgup2!
We show pretraining on task data alone suffices to bridge the gap between state space models and transformers on Long Range Arena, leading to a significantly better estimate of model capabilities.
arxiv.org/abs/2310.02980
🧵
Ankit Gupta retweeted
Helping or Herding? Reward Model Ensembles Mitigate but do not Eliminate Reward Hacking
paper page: huggingface.co/papers/2312.0…
Reward models play a key role in aligning language model applications towards human preferences. However, this setup creates an incentive for the language model to exploit errors in the reward model to achieve high estimated reward, a phenomenon often termed reward hacking. A natural mitigation is to train an ensemble of reward models, aggregating over model outputs to obtain a more robust reward estimate. We explore the application of reward ensembles to alignment at both training time (through reinforcement learning) and inference time (through reranking). First, we show that reward models are underspecified: reward models that perform similarly in-distribution can yield very different rewards when used in alignment, due to distribution shift. Second, underspecification results in overoptimization, where alignment to one reward model does not improve reward as measured by another reward model trained on the same data. Third, overoptimization is mitigated by the use of reward ensembles, and ensembles that vary by their pretraining seeds lead to better generalization than ensembles that differ only by their fine-tuning seeds, with both outperforming individual reward models. However, even pretrain reward ensembles do not eliminate reward hacking: we show several qualitative reward hacking phenomena that are not mitigated by ensembling because all reward models in the ensemble exhibit similar error patterns.
Ankit Gupta retweeted
[4/4] Investigating the effects of data scale, we find self-pretraining is most effective in low-data regimes, underscoring its importance for evaluation across all dataset sizes. We further show that self pretraining is effective across model sizes and when compute is limited.
Ankit Gupta retweeted
[3/4] The marked effect of self-pretraining on long-sequence tasks leads us to rethink the necessity of complex designs, with Diagonal Linear RNNs (DLR) as a specific example. Our findings indicate that, when pretrained, simple architectures can be as effective as complex designs
Ankit Gupta retweeted
[2/4] By self-pretraining state space models and transformers with denoising objectives on individual Long Range Arena tasks, we found both models achieve similar performance and surpass a trained from scratch S4, with Transformer surpassing all previous results on text-tasks.
Ankit Gupta retweeted
Excited to share my work with @JonathanBerant @ankgup2!
We show pretraining on task data alone suffices to bridge the gap between state space models and transformers on Long Range Arena, leading to a significantly better estimate of model capabilities.
arxiv.org/abs/2310.02980
🧵
[NEW] work with @AmosaurusRex @JonathanBerant Transformers are very successful & yet they reportedly struggle on tasks with sparse supervision e.g. Long Range Arena. Pretraining them on the task data itself allows Transformers & simpler SSMs to match sophisticated SSMs like S4!!
Excited to share my work with @JonathanBerant @ankgup2!
We show pretraining on task data alone suffices to bridge the gap between state space models and transformers on Long Range Arena, leading to a significantly better estimate of model capabilities.
arxiv.org/abs/2310.02980
🧵












