Skip to content
InfoResearchPreprint

TraceRouter++: Causal Semantic Routing Realignment via Optimal Transport

Published
Record updated
View JSON

Summary

TraceRouter++ is a routing-level safety framework for large foundation models that traces causal routing pathways activated by harmful inputs and realigns their unsafe activation states. It uses sparse autoencoders, feature influence scores, and a constrained optimal transport soft realignment instead of rigid neuron suppression. The authors report improved adversarial robustness and a favorable safety-utility trade-off across LLMs, MLLMs, and diffusion models.