179 questions
No questions match those filters.
How does the router learn, given that top-K is not diff...
This is one of the questions in the full AI/ML interview bank. Pro unlocks all 1789 questions; Premium includes the same bank plus the highest daily Practice limit.
See plansWhile the top-K selection itself is not differentiable, the gate weights are differentiable with respect to the router embeddings. This allows the model to learn how to rank experts correctly for specific tokens. A significant risk is that experts never selected by the router receive no gradient signal and fail to improve. Without auxiliary load-balancing losses, the model may collapse into a state where one or two experts handle all traffic, causing the remaining experts to atrophy and effectively turning the model into a single-expert system with wasted parameters.