We propose SmoothDiff, a novel feature attribution method leveraging automatic differentiation, directly targeting nonlinearities responsible for the shattered gradient problem.

We accelerate Taylor mode for practically relevant differential operators by collapsing Taylor coefficients; this can be done automatically with compute graph simplifications