GrAInS Uses Integrated Gradients for Inference-Time Token Steering

GrAInS enables inference-time steering for LLMs and VLMs by using Integrated Gradients to identify top-k influential tokens. This method moves beyond fixed intervention vectors by using contrastive, gradient-based attribution to modify internal activations based on specific token contributions to preferred or dispreferred outputs.