Can you release a model's weights openly and still control who gets access to its more sensitive capabilities?
Tiered Language Models keep those capabilities in the weights but gate access behind a secret key.
Check out this approach to safe open-weight release in our paper:
Can we release all the weights of an LLM but still provide differential access to privileged users?
Yes! We introduce: ๐ง๐ถ๐ฒ๐ฟ๐ฒ๐ฑ ๐๐ฎ๐ป๐ด๐๐ฎ๐ด๐ฒ ๐ ๐ผ๐ฑ๐ฒ๐น๐ (๐ง๐๐ ๐). Define access tiers corresponding to different computation graphs over the same set of LLM parameters!
Excited to share our new paper!
โForecasting Downstream Performance of LLMs With Proxy Metricsโ
w/ my amazing advisors @sivareddyg, @mariusmosbach, @DBahdanau
Cross-entropy loss is a poor predictor of how models perform on downstream tasks (esp. reasoning). We propose something better: proxy metrics computed over expert reasoning traces.
๐งต Thread below ๐
SAEs fail at OOD tasks. Why?
Features in superposition are linearly representable but not linearly accessible. Instead of discarding sparse coding, we embrace the geometry of superposition and use methods equipped to handle the nonlinearity it induces.
Mechanistic interpretability aims to understand models โ and the more superhuman or incoherent they become, the more we need that understanding to be reliable. We propose a framework for this, drawing on established tools from causal reasoning and statistical identifiability:
๐งต
Checkout our latest work on building self-supervised text embeddings without relying on contrastive data. โ๏ธ
The main idea behind LLM2Vec-Gen is trying to encode a model's answer to a query, rather than the query itself.
Your LLM already knows the answer. Why is your embedding model still encoding the question?
๐จIntroducing LLM2Vec-Gen: your frozen LLM generates the answer's embedding in a single forward pass โ without ever generating the answer. Not only that, the frozen LLM can decode the embedding back into text.
๐ SOTA self-supervised embeddings
๐ก๏ธ Free transfer of instruction-following, safety, and reasoning
LLM2Vec-Gen represents a major paradigm shift for embeddings/retrieval. Why encode the query when the LLM already knows what to look for and can directly produce an embedding for it?
Best part: itโs self-supervised, and it does all of this while the LLM remains completely frozen.
Think about it: "solve xยฒ + 3x โ 4 = 0" has zero reasoning in it. But the LLM's response does. By encoding the response, the embedding captures the reasoning --- and the better the LLM reasons, the better the embedding. This is why our results scale with model size. As LLMs get smarter, our embeddings automatically get better.
LLM2Vec-Gen is also the first demonstration of the promise of @ylecun's JEPA for text embeddings. The alignment loss is JEPA โ predict in representation space, not token space. The reconstruction loss goes beyond --- it keeps embeddings decodable.
This paradigm shift opens new frontiers:
๐ฌ Can we build a full JEPA for language where the teacher and student are the same LLM?
โก Can LLMs reason in compressed space without ever generating text?
๐ค Can agents reason in compression tokens and carry that directly into retrieval?
๐ฌ Can agents talk to each other in compression tokens instead of text --- dense, fast, and still human-readable?
LLM2Vec-Gen is a first step toward all four.
I wrote my first ever blog post! "Agentic Coding: A New Abstraction Layer in the Programming Stack"
I give some thoughts about how my coding changed with agents, where I think it's headed, and how the resistance to adopting them echoes past shifts in CS.
Link below๐
first paper of the phd ๐ฅณ
the Superficial Alignment Hypothesis (SAH) argues that pre-training adds most of the knowledge to a model, and post-training merely surfaces it.
however, this hypothesis has lacked a precise definition. we fix this.
๐จNew paper
Are visual tokens going into an LLM interpretable ๐ค
Existing methods (e.g. logit lens) and assumptions would lead you to think โnot muchโ...
We propose LatentLens and show that most visual tokens are interpretable across *all* layers ๐ก
Details ๐งต
๐จHow do LLMs acquire human values?๐ค
We often point to preference optimization. However, in our new work, we trace how and when model values shift during post-training and uncover surprising dynamics.
We ask: How do data, algorithms, and their interaction shape model values?๐งต
Itโs clear next-gen reasoning LLMs will run for millions of tokens. RL at 1M needs ~100ร compute than 128K. Our Markovian Thinking keeps compute scaling linear instead. Check out Miladโs thread; some of my perspectives below:
Introducing linear scaling of reasoning:
๐๐ก๐ ๐๐๐ซ๐ค๐จ๐ฏ๐ข๐๐ง ๐๐ก๐ข๐ง๐ค๐๐ซ
Reformulate RL so thinking scales ๐(๐ง) ๐๐จ๐ฆ๐ฉ๐ฎ๐ญ๐, not O(n^2), with O(1) ๐ฆ๐๐ฆ๐จ๐ซ๐ฒ, architecture-agnostic.
Train R1-1.5B into a markovian thinker with 96K thought budget, ~2X accuracy ๐งต
i will be presenting AgentRewardBench at
#COLM2025 next week!
session: #3
date: wednesday 11am to 1pm
poster: #545
come learn more about the paper, my recent works or just chat about anything (montreal, mila, etc.)
here's a teaser of my poster :)
๐จ New paper!
LLMs, when asked harmful questions, sometimes produce outputs that look helpful (and harmful) โ but are actually ๐ฑ๐ฒ๐น๐ถ๐ฏ๐ฒ๐ฟ๐ฎ๐๐ฒ๐น๐ ๐๐ฟ๐ผ๐ป๐ด
Whatโs bad - current LLM-based jailbreak scorers canโt tell the difference (me neither)
More in ๐งต๐
If you're interested in working on agent safety (and are a student in Canada) you should apply to this! @gspandana is extremely smart and one of the kindest people I've gotten to work with
Internship @ServiceNowRSRCH to build the next generation of computer use agents that
are safe and secure from malicious attacks. Focus on intervention strategies, defenses to make agents robust against unsafe behavior..
Apply here: https://t.co/XPBMl8SaVs
๐จ Incredibly excited to share that I'm starting my research group focusing on AI safety and alignment at the ELLIS Institute Tรผbingen and Max Planck Institute for Intelligent Systems in September 2025! ๐จ
Hiring. I'm looking for multiple PhD students: both those able to start in Fall 2025 (i.e., as soon as possible) and through centralized programs like CLS, IMPRS, and ELLIS (the deadlines are in November) to start in SpringโFall 2026. I'm also searching for postdocs, master's thesis students, and research interns. Fill the Google form below if you're interested!
Research group. We will focus on developing algorithmic solutions to reduce harms from advanced general-purpose AI models. We're particularly interested in alignment of autonomous LLM agents, which are becoming increasingly capable and pose a variety of emerging risks. We're also interested in rigorous AI evaluations and informing the public about the risks and capabilities of frontier AI models. Additionally, we aim to advance our understanding of how AI models generalize, which is crucial for ensuring their steerability and reducing associated risks. For more information about research topics relevant to our group, please check the following documents:
- International AI Safety Report,
- An Approach to Technical AGI Safety and Security by DeepMind,
- Open Philanthropyโs 2025 RFP for Technical AI Safety Research.
Research style. We are not necessarily interested in getting X papers accepted at NeurIPS/ICML/ICLR. We are interested in making an impact: this can be papers (and NeurIPS/ICML/ICLR are great venues), but also open-source repositories, benchmarks, blog posts, even social media postsโliterally anything that can be genuinely useful for other researchers and the general public.
Broader vision. Current machine learning methods are fundamentally different from what they used to be pre-2022. The Bitter Lesson summarized and predicted this shift very well back in 2019: "general methods that leverage computation are ultimately the most effective". Taking this into account, we are only interested in studying methods that are general and scale with intelligence and compute. Everything that helps to advance their safety and alignment with societal values is relevant to us. We believe getting thisโsome may call it "AGI"โright is one of the most important challenges of our time.
Join us on this journey!
What's the path to scalable and safe web agents? Is web agents the new semantic parsing? I will be giving a talk at the ACL REALM workshop today at 9:30 am. Come check out if you are interested in the history and contemporary work in this area. Lot of other exciting speakers. #ACL2025 #ACL2025NLP
https://t.co/RNk5TPjDea
Come by our #ACL2025 poster tomorrow to discuss the safety risks surrounding increasingly capable instruction-following retrievers (or anything safety related)!
16:00-17:30 on Tuesday in Hall 4/5
Come and visit our poster on the Safety of Retrievers @aclmeeting
๐๏ธTuesday, Findings Posters, 16:00-17:30
๐จInstruction-following retrievers will become increasingly good tools for searching for harmful or sensitive information.๐จ