Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
We introduce Cross-Interface Policy Regularization (CIPR), a self-supervised objective for LLM agents that enforces consistent action distributions across semantically equivalent but syntactically different interfaces. By penalizing KL divergence between policies conditioned on randomized action identifiers, CIPR mitigates the brittle dependence of agent competence on specific interface formulations. Evaluation shows CIPR significantly improves robustness to out-of-distribution interface perturbations, recovering 10.5% accuracy in agentic function calling tasks while maintaining performance on the original interface.
LLM agents trained on function calling forget nearly 25% of their capability when action names are perturbed, but a simple self-supervised regularization recovers over half of that loss.
Post-training binds the competence of an LLM agent to the interface it was trained on. Semantics-preserving perturbations of that interface, such as renaming an action or changing the notation, leave the underlying task unchanged yet make the agent worse at it. Scoring each sample independently means interface randomization only broadens this dependence from a single interface to the training interface distribution, so out-of-distribution perturbations still degrade the agent. We propose Cross-Interface Policy Regularization (CIPR), a self-supervised objective that replaces action names with uninformative identifiers and presents each training sample under two interfaces in which equivalent actions are identified by construction. CIPR penalizes the KL divergence between the action distributions induced under the two interfaces, so the policy behaves consistently across them. We evaluate CIPR in both supervised and reinforcement-learning post-training, where it improves robustness. On agentic function calling, held-out interface perturbations reduce the accuracy of a post-trained agent by nearly 25%, and CIPR recovers 10.5% while leaving accuracy on the unchanged interface intact.
UW