ML PhD student @TechnionLive working with @HaggaiMaron

Israel
Joined December 2021
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🧵
1
3
6
393
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. 🧵
4
21
2
98
7,319
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.
6
17
2
100
29,670
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.
1
1
9
657
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
3
20
2,198
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]
2
7
2
11
2,135
🧵"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
1
14
2
75
9,035
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
2
30
5,332
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
39
261
74
1,773
473,000
Come check out our paper🌴
1/6 Better late than never! 🚀 Check out our paper GradMetaNet at NeurIPS San Diego *today* at 4:30 PM. We take a weight-space learning approach to gradient-based model adaptation (editing, learned optimization) using a rank-1 decomposition of gradients and their symmetries.
2
260
📢 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
1
6
2
28
7,475
Topological blindspots is an ICLR oral presentation! Come check it out 🔥
🍩 Topological blindspots is coming to ICLR as an oral presentation! 🍩 We prove that message-passing based topological deep learning (TDL) architectures are unable capture basic topological invariants including homology, orientability, planarity and more.
2
16
885
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
10
27
2,574
🧠 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…
3
18
74
5,408
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]
6
385
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)
18
177
14
1,739
153,488
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
1
8
1,353