
ML PhD student @TechnionLive working with @HaggaiMaron
Israel
Joined December 2021
- Tweets32
- Following205
- Followers363
- Likes62
Yam Eitan retweeted
1/8 Can a neural network learn to compress other neural networks?
New paper with @ytn_ym and @HaggaiMaron, in which we take a first step toward this goal with CrossGMN, a method that learns to transform trained models into different architectures directly in weight space.
Up to an 8.89× speedup in distillation steps. Here’s the idea🧵
Yam Eitan retweeted
I'm hiring a postdoc! The position is funded by my new ERC Starting Grant, GDLNeuralArt, and based in my group at the Technion.
We work on learning on weight spaces, internal representations and gradient spaces: treating neural networks themselves and their artifacts as data. 🧵
1/
How much can you compress an LLM’s KV cache?
tl;dr it depends on how you train your model.
Many strong context compaction methods, such as Cartridges and attention matching, operate post-hoc: given a fixed model and a context, they try to compress the resulting KV cache.
@yoav_gelberg and I ask the complementary question:
can we train the model to produce KV representations that are easier to compress?
In other words: keep the compression method fixed, and change the representations it sees.
5/
Empirically, KV-CAT consistently improves downstream KV compression quality across multiple settings.
We evaluate over multiple model sizes and KV-compaction methods:
• suffix perplexity under compressed prefixes
• long-context retrieval
• long-context QA
Key findings:
• 3.21x better perplexity retention under compression
• 5x faster optimization for gradient-based compressors
• 68% better retrieval accuracy
• 39% improvement in LongBench QA accuracy under compressed context.
Overall, the results suggest that compression-aware training can make transformers fundamentally more amenable to post-hoc KV compression.
6/
Check out the full paper: arxiv.org/abs/2605.05971
Code coming soon.
Huge thanks to our collaborators @yoav_gelberg* (co-first), @HaggaiMaron, @yaringal and @mmbronstein
for making this project happen.
This work was heavily inspired by recent progress in KV cache compaction by @AdamZweiger @EyubogluSabri @ryansehrlich @simran_s_arora
🤖 Made with AI
Yam Eitan retweeted
New blogpost out 📃
"Detecting LLM Misbehaviors from the Inside Out with Deep Learning on Structured Data" (ffabffrasca.substack.com/p/d…) [1/8]
Yam Eitan retweeted
🧵"Neural Message Passing on Attention Graphs for Hallucination Detection" at #ICLR2026 !
🕸️We apply GNNs on the structured data LLMs produce as they generate text (e.g. attentions) to predict their errors.
📄 arxiv.org/abs/2509.24770
🤝 @GuyBarSh (co-1st) @YftahZ @HaggaiMaron
How powerful are models that operate directly on weights? Check out the answer in our new preprint!
How powerful are neural networks that learn from other neural networks?
New paper with @ytn_ym and Prof. @HaggaiMaron, in which we provide a comprehensive theoretical foundation for weight-space networks.
📄 arxiv.org/pdf/2602.01083
Yam Eitan retweeted
Introducing DroPE: Extending the Context of Pretrained LLMs by Dropping Their Positional Embeddings
pub.sakana.ai/DroPE/
We are releasing a new method called DroPE to extend the context length of pretrained LLMs without the massive compute costs usually associated with long-context fine-tuning.
The core insight of this work challenges a fundamental assumption in Transformer architecture. We discovered that explicit positional embeddings like RoPE are critical for training convergence but eventually become the primary bottleneck preventing models from generalizing to longer sequences.
Our solution is radically simple: We treat positional embeddings as a temporary training scaffold rather than a permanent architectural necessity.
Real-world workflows like reviewing massive code diffs or analyzing legal contracts require context windows that break standard pretrained models. While models without positional embeddings (NoPE) generalize better to these unseen lengths, they are notoriously unstable to train from scratch.
Here, we achieve the best of both worlds by using embeddings to ensure stability during pretraining and then dropping them to unlock length extrapolation during inference. Our approach unlocks seamless zero-shot context extension without any expensive long-context training.
We demonstrated this on a range of off-the-shelf open-source LLMs. In our tests, recalibrating any model with DroPE requires less than 1% of the original pretraining budget, yet it significantly outperforms established methods on challenging benchmarks like LongBench and RULER.
We have released the code and the full paper to encourage the community to rethink the role of positional encodings in modern LLMs.
Paper: arxiv.org/abs/2512.12167
Code: github.com/SakanaAI/DroPE
Yam Eitan retweeted
📢 Introducing:
Learning on LLM Output Signatures for Gray-box LLM Behavior Analysis [arxiv.org/pdf/2503.14043]
A joint work with @ffabffrasca (co-first author) and our amazing collaborators:
@dereklim_lzh @yoav_gelberg @YftahZ @el_yaniv @GalChechik @HaggaiMaron
🧵Thread
Topological blindspots is an ICLR oral presentation! Come check it out 🔥
Yam Eitan retweeted
"Balancing Efficiency and Expressiveness: Subgraph GNNs with Walk-Based Centrality" by @JoshSouthern13, @ytn_ym, Guy Bar-Shalom, @mmbronstein, @HaggaiMaron, @ffabffrasca
Paper: arxiv.org/abs/2501.03113
#graphneuralnetworks
Yam Eitan retweeted
🧠 Can we train models to manipulate NN weights?
Join us at the first Weight Space Learning Workshop @ICLR_conf to explore learning directly from neural network parameters!
📝 CfP: weight-space-learning.github…
⏰ Deadline: Feb 4, 2025
🌐 More Info: weight-space-learning.github…
Check out our new paper!
🎉``A Flexible, Equivariant Framework for Subgraph GNNs via Graph Products and Graph Coarsening'' is accepted to #NeurIPS2024!🎉
➡️arxiv.org/pdf/2406.09291
This is a joint effort with @ytn_ym and was made possible by amazing collaborators: @ffabffrasca @HaggaiMaron
[1/8]
Yam Eitan retweeted
Topological machine learning and topological data analysis are two of the most interesting and fast-growing fields if you're interested in cool maths and machine learning. (1/3)
Extremely honored to be on this list! Thanks @miniapeur for the support!
Replying to @miniapeur
(2/3) There are a lot of extraordinary researchers in these two areas, and I wanted to give them a little publicity (in no particular order). Follow them and check out their work: @Pseudomanifold @ninamiolane @rballeba @ClaBat9 @hansmriess @cutezu_ @mathildepapillo @lorgiusti @ytn_ym @mmbronstein @LidaKanari @HajijMustafa @maosheng_yang @KrishnaswamyLab @crisbodnar
























