grad_reg Training using gradient regularization (https://ojs.aaai.org/index.php/AAAI/article/view/11504)