Learning interaction rules from multi-animal trajectories via augmented behavioral models