For USCIS benefits, you can legally enter any combination of the names on the Marriage Certificate. You can enter the new legal name on I-485, etc.
Example: John Smith marries Jane Doe.
Jane can use Jane Doe, Jane Smith, Jane Doe Smith, or Jane-Doe-Smith as new legal name on all USCIS paperwork, and she can use that as her legal name.