Unify the interface for robust multinomial and revise documentation

Pivotal ID: 52350975