doodle (@doodlext25) profile picture

CS PhD student @HKUniversity, supervised by @difanzou.

Joined November 2024
Photo by doodle (@doodlext25)Photo by doodle (@doodlext25)Photo by doodle (@doodlext25)Photo by doodle (@doodlext25)Photo by doodle (@doodlext25)
Interesting finding from Mistral Large 4: PPL correlates well with knowledge-based evals, but much less so with agentic performance (SWE-bench Verified). If PPL isn't a good predictor of agentic potential, what should we be looking at instead? 🤔
Today, we release the weights of a new and improved Le Chonk, advancing the open frontier for agentic coding and cybersecurity. 💻 Built to build: Le Chonk ranks #1 on @cognition’s FrontierCode among open models, and sets a new benchmark for Pareto efficiency on @artificialanalysis. 🛡 Built to defend: State-of-the-art in cyberdefense, put through the paces by the world's leading cyber vendors, including @CrowdStrike, @cloudflare, @rubrik, @zscaler, and others. Built for sovereignty: Make it yours. Download the weights on @huggingface , customize it with Mistral Forge, and deploy it wherever you need it. Read more from our team and customers 👇:
27
doodle retweeted
Muon now trains frontier LLMs. But how does its spectral orthogonalization shape feature learning? New paper: on factual recall, Muon learns facts in a more balanced way and is far less slowed by softmax saturation. 🔗arxiv.org/abs/2610.02798 [1/8]
10
16
3
127
5,868
(1/6) Weight decay has a direction, too. We usually tune its strength. For matrix weights, we can also choose how shrinkage is distributed across singular modes. New blog with my advisor @difanzou : Rethinking Weight Decay doodlext.github.io/blog/reth…
6
1
1
9
947
(6/6) Implementation: integer powers need no SVD. Odd powers use Gram products; even powers also use a polar approximation. The broader question: how should we distribute regularization across modes, modules, and spectral bands? I'd love to hear what you would test next.
50
(5/6) Nearly the same pretraining loss, different internal geometry. Ordinary WD vs asymmetric decay (p=0 Attention, p=2 MLP): Loss gap: 1.47e-5 nats MLP weight stable rank: +40.3% Similar prediction loss leaves room for different spectral structures.
51
(4/6) Case study: head-wise Muon. Updating V as a full matrix improves validation loss over splitting V by head. Adding p=2 decay on V further improves the default full-V recipe: 3.01968 -> 3.01596 -> 3.01324 Update geometry and decay geometry offer two tuning choices.
50
(3/6) Does this spectral preference survive training? We start with an exactly solvable linear model, then examine an 8-layer Transformer. The spectrum-shaping effect remains visible after task-driven training, with different responses across matrix types.
77
(2/6) Spectral-power WD replaces W with: S_p(W) = U diag(sigma_i^p) V^T Under pure decay: p=0: stronger relative pressure on small modes p=1: ordinary WD; spectral ratios stay fixed p=2: stronger relative pressure on large modes One exponent, different spectral preferences.
192