The federal government is in talks with Citigroup Inc. to increase its ownership stake in the troubled bank.