Dylan Jayabahu — AI researcher, CS at Waterloo.

Dylan Jayabahu

CS @ Waterloo · AI research

1.000
same points, opposite reading

Research

01 The Truth Was Never Gone: Perfect Aliasing in Compliant-Context Truth Probes NeurIPS 2026 Interpretability as a ScienceSole author

Truth probes fall below chance on models trained to deceive, and the field reads that as the model hiding what it knows. It cannot be. On the compliant data those probes are fit on, belief, intent, and emitted token are one label vector, so a “truth probe” and an “action probe” are a single fitted direction scored against opposite labels. Nothing had to be hidden.

0.006
Conventional probe · AUROC
1.000
Identified probe · same activations
751
Cell-layer pairs · identity exact
The same 751 (cell, layer) pairs, twice. On compliant data the truth and action AUROCs are one fitted direction scored against opposite labels: every pair sits on action = 1 − truth, to one machine epsilon. On mixed contexts the collinearity breaks and the identity is gone. Measured, across 39 cells.

The aliasing rate then follows in closed form from two published behavioural rates and no activations at all, released as a 40-line audit function. The constructive half is a randomized-codebook task plus mixed-context fitting: on identical activations of a Gemma-2-9B policy lying on 100% of adversarial trials, the conventional probe reads 0.006 while the identified probe reads 1.000 with zero seed variance, holding at the final layer in 19 of 21 cells across four architectures (3B to 32B). Scoped to linear decodability, not causal use.

02 Off-Axis Drift: Internalizing a Steering Direction Needs More Than Its Scalar Projection Interpretability · Efficient ReasoningFirst author, with Tinuade Adeleke

Interpretability keeps producing read-then-steer directions (refusal axes, value axes, truthfulness features) that only work at inference, behind a hook. Moving one into the weights turns out to demand the whole steered activation: matching its scalar projection, the target prior work adopted untested, does not install the behaviour at any scale tested.

22.6%
Thinking cut · held-out set
0 of 6
Scalar arms holding accuracy
3
Model scales · 1.5B to 14B
Why a scalar is not enough. Each arm plotted by how far it moved along the halt axis, against how much it disturbed everything else. The two families never overlap — the shaded wedge is empty by construction, because satisfying one scalar leaves every other dimension free. Comparative only: no arm leaves the base model’s natural off-axis range.

Across Qwen 1.5B/7B/14B, no matched-capacity scalar arm compresses at held accuracy, while full-vector reconstruction cuts 22.6–33.1%. The mechanism is off-axis drift: satisfying one scalar leaves every other dimension free, and the frozen downstream reader depends on those. The direction itself is given a mechanistic identity, since projecting it out adds ~1,480 think-tokens and collapses termination by 23 points, its cosine overlap with the end-of-think logit direction is 0.004 (so it is not a disguised “emit </think>” push), and the effect is ~96% reader-mediated and localized to the final two MLP layers. A pre-registered second-direction test then bounded the claim: a near-orthogonal refusal direction installed under both targets, so the universality claim was struck from the main text rather than softened.

03 The Halt Vector: Internalizing a Causal Steering Intervention for Efficient Reasoning COLM 2026 Workshop on Efficient ReasoningFirst author, with Tinuade Adeleke

A reasoning model reaches its answer roughly twice as early as it stops writing. Found a causal “halt vector” in the residual stream and internalized it into the weights: a hook-free, per-problem self-halt that cuts overthinking without losing accuracy.

24%
Less thinking, held accuracy
0.70
Adaptivity (cut vs. slack)
5
Unseen benchmarks

Difference-of-means halt vector at layer 18 of DeepSeek-R1-Distill-Qwen-7B. Steering it controls generation length monotonically (ρ = −0.82) where a replicated value axis is inert (+0.09). Internalized by reconstructing the full steered activation with the off-axis dimensions pinned; the scalar-projection target backfires, running 43% longer at 0.57 accuracy. Fit from 24 problems, no RL; stable across 3 seeds; closes an AIME cap-loop pathology (31–38% → 6–9%). That scalar-versus-full-vector question is taken up at scale in Off-Axis Drift, above.

04 Watching a Reward Hack Form InterpretabilitySole author · UWaterloo, 2026

Dense LoRA checkpointing turns a single GRPO run into a frame-by-frame movie of a 7B reasoning model's internals, so a failure mode can be caught as it crystallizes, not just post-hoc.

3
Seeds reproduced
15–30steps
Length leads collapse
1×H200
Single-GPU pipeline

GRPO/RLVR + dense LoRA checkpointing + per-step probing. Reasoning-length compression leads the accuracy collapse by 15–30 steps: an early-warning signal the model’s own confidence misses (it stays ~0.9 as accuracy halves). An α-sweep maps a tunable critical penalty: a phase boundary, not a ramp.

05 Deep Learning Model for Invasive Ductal Carcinoma Detection in Histopathology Images IEEE CCECE 2025Sole author · Vancouver, May 2025

A computer-aided diagnosis pipeline for invasive ductal carcinoma that pairs sliding-window heatmaps with a homogeneous-region oversampling technique, preserving biological structure while correcting class imbalance. State-of-the-art on the public benchmark.

89.06%
Balanced accuracy
86.68%
F1 score
SOTA
vs. prior literature

Interpretable heatmap visualization makes the model a clinical-grade second reader for pathologists rather than a black box. Indexed on IEEE Xplore.

06 A Real-World mmWave Radar Dataset for Human Action Detection JCVIS · CVIS 2024First author, with Parthipan Siva · UWaterloo, 2024

The first FMCW mmWave dataset captured in natural living environments rather than a lab setup, positioning privacy-preserving radar as a viable backbone for aging-in-place monitoring. CNN baselines establish where the real difficulty lies.

28
Residences
458
Sit-down events
454
Stand-up events

TI IWR6843AOP FMCW sensor · 3D point clouds · 3-layer CNN with sliding-window detection. Presented at CVIS 2024.

Projects

Play

Sit down at the table

No-Limit Hold’em · 6-max · Blinds $5/$10