Thoughts on SSM conv1d patch?

#6
by bfkxnr - opened

There is a thread on the base model discussion page where LuffyTheFox shared some insights into the anomaly in late attention layers where conv1d scales look significantly inflated. Redashes then confirmed the findings and proposed a scale factor matrix to "cool down" these tensors: redashes/Qwen3.8-27B-BF16-SSMFIX. According to the model page:

Without repair, the drifted scales let the recurrent state saturate/collapse: long-context (75k+) collapse, repetition loops, mid-generation truncation, and "philosophizing" drift where the model abandons the task. Short-context perplexity looks normal → silent degradation.
TruthfulQA generation up +6~8pp across the board → strong hallucination reduction

ThinkingCap carries the same inflated values in ssm_conv1d.weight as the base model and I was wondering if it could potentially benefit from the patch so we can get the best of both worlds: reduced thinking + preventing recurrent state from running hot. Luckily, the patch is very simple, just multiply the conv1d weights in 8 layers using the rescale matrix on redashes' page. Back up the original weights so that you can revert safely.

Curious to hear your thoughts on this. I will be trying it out myself in the meantime in a very unscientific way 🥸

Sign up or log in to comment