A bit overdue, but "Do Thinking Tokens Help with Safety?" was selected as the ๐ Best Paper at the ICML 2026 AI4GOOD workshop!
The prize money will, of course, be put to good use (i.e., API credits) ๐
(And all credits to @qw3rtman for taking the great photos!)
Surprising finding about AI safety from our group. We find that in small open-parameter models, the thinking tokens have almost no effect on safety. On unsafe prompts, the refusal/compliance is predictable from a simple trained head given the prompt alone. Great work and thread by @narutatsuri
What happens when AI is guided by law-like principles? Can we design some computational tools to "debug" rules?
Check out our new work ๐ ๐๐ฅ๐๐ฅ๐ฆ๐ฅ๐ ๐ฃ๐ช โ๐ ๐๐ค๐ฅ๐ฃ๐ฆ๐๐ฅ๐๐ ๐ ๐๐๐ ๐๐๐ฅ๐๐ฃ๐ก๐ฃ๐๐ฅ๐๐ฅ๐๐ ๐ ๐๐ ๐ฃ ๐ธ๐๐งโโ๏ธ to find out more!
๐งต(1/10)
First paper of my PhD with my amazing advisors!
Thereโs been a ton of hype and media coverage on OpenEvidence as an โAI co-pilot for cliniciansโโฆ and our long-horizon benchmark puts them to the test!! Our results suggest they are far from reliable for downstream use.
Robot policies can move but can't think.
LLMs can think but can't move.
So we connected them.
Real robot: 16.7% โ 97.3%
Sim (LIBERO-PRO): 12.8% โ 53.3%
While FIFA has captured everyoneโs attention, something consequential happened in academia this week, especially for international students in PhD programs.
DHS issued a final rule changing how long F-1 students may remain in the United States.
A thread on what it means and what students and universities can do. ๐งต
A bit overdue, but "Do Thinking Tokens Help with Safety?" was selected as the ๐ Best Paper at the ICML 2026 AI4GOOD workshop!
The prize money will, of course, be put to good use (i.e., API credits) ๐
(And all credits to @qw3rtman for taking the great photos!)
Surprising finding about AI safety from our group. We find that in small open-parameter models, the thinking tokens have almost no effect on safety. On unsafe prompts, the refusal/compliance is predictable from a simple trained head given the prompt alone. Great work and thread by @narutatsuri
New Paper: Rethinking On-Policy Self-Distillation for Thinking Models
On-policy self-distillation (OPSD) is a promising approach to recursive self-improvement. Thinking models seem especially well positioned to teach themselves with privileged information since they can deliberate, verify, and self-correct.
Surprisingly, we find that OPSD can make thinking models worse.
(More details in ๐งตbelow)
Itโs deployment time!
Youโve done the pre-deployment evals. You THINK your model is safe, so you ship it ๐
๐จ After deployment, reports of misbehavior start trickling in
What happened?? How could you have caught it?? ๐ค
@icmlconf 2026 Spotlight!
๐งต
A question on synthetic data generation: If we want a language model to solve k-step arithmetic problems (such as a+b*c-d=?), with operands from 1 to 100, which training distribution should we use?
A. Uniform distribution: Sample these k operands uniformly from 1 to 100
B. Power law: randomly shuffle 1-100 and impose an artificial power law. Sample these k operands according to this power law.
โกOur ICML 2026 (spotlight) paper shows: Option B is better! Surprisingly, the same idea extends far beyond this simple example to many reasoning tasks that require implicit composition of multiple atomic skills, including multi-hop QAs and synthetic GSM problems.
๐Paper: https://t.co/8LVlmCCBz9
๐Blog: https://t.co/t2yhIJ0s1P
Surprising finding about AI safety from our group. We find that in small open-parameter models, the thinking tokens have almost no effect on safety. On unsafe prompts, the refusal/compliance is predictable from a simple trained head given the prompt alone. Great work and thread by @narutatsuri
Iโm also at #ICML2026 and will be giving an Oral presentation at the ICML 2026 AI4GOOD Workshop (https://t.co/kdmkHSdp8t)!
This will be on July 10th 13:30โ14:45 at Grand Ballroom 103. Please stop by if you're around and have time!
Iโll be around all week, so please feel free to reach out if you want to chat about AI safety and/or (mis)alignment!
New Paper: Do Thinking Tokens Help with Safety?
Reasoning models generate โthinkingโ tokens before answering, which improves performance on many tasks like math and coding.
A natural hope is that models can also use thinking to deliberate about whether to refuse or comply a request.
But how true is this?
We find a surprisingly sobering answer: current reasoning models often do not use thinking tokens for real safety deliberation.
Work with: Abhishek Panigrahi, @prfsanjeevarora at @PrincetonPLI
(More in ๐งตbelow)
(5/n) Finally, we study how existing inference-time and training-based defenses change model behavior: while many are motivated to โactivate safety reasoning inside the thinking trace,โ do they actually improve the ASR-ORR trade off and/or improve deliberation signals?
We find that most do not: reducing harmful compliance often comes at the cost of more over-refusal, and current methods actually suppress the already scarce deliberation signals in the thinking trace instead of increasing them.
New Paper: Do Thinking Tokens Help with Safety?
Reasoning models generate โthinkingโ tokens before answering, which improves performance on many tasks like math and coding.
A natural hope is that models can also use thinking to deliberate about whether to refuse or comply a request.
But how true is this?
We find a surprisingly sobering answer: current reasoning models often do not use thinking tokens for real safety deliberation.
Work with: Abhishek Panigrahi, @prfsanjeevarora at @PrincetonPLI
(More in ๐งตbelow)
(3/n) Next, we ask whether the visible thinking trace contains real safety deliberation.
We use frontier models to annotate each sentence in a reasoning trace and identify โoscillationsโ: places where the model appears to deliberate between refusal and compliance. At these points, we roll out many final responses to test whether the text-level deliberation actually changes the final refusal/compliance distribution.
Result: most visible deliberation signals have no meaningful effect on the final decision (grid-shaded region in right bars).