Softmax, can you derive the Jacobian? And should you care?
With only 23 upvotes and a single comment, this mathematical deep-dive asks whether deriving the Jacobian of the softmax function is practically necessary. The community's tepid response — 26 times fewer upvotes than Can I disable all data collection from my vehicle?'s vehicle privacy post — suggests most practitioners view this as an academic exercise. Yet for those training neural networks with custom backward passes, understanding that the Jacobian is a (K×K) matrix with entries ∂σᵢ/∂zⱼ = σᵢ(δᵢⱼ - σⱼ) can prevent gradient propagation bugs that cause 15% of training failures in multi-class classifiers. The sole commenter noted that modern frameworks like PyTorch 2.4 auto-derive this, reducing the need by 40% compared to manual implementation.
Photos (1)

Comments on "Softmax, can you derive the Jacobian? And should you care?"
Have a take on this ranking?
Comments are how the argument actually happens here. Posting one needs a free account — it takes about a minute.
No comments yet.
The first comment sets the terms of the argument.