I don't know what model headset and/or cartridge bearings Trek used as OEM but there are some very good Integrated/Zero stack headset bearing out there. Cane Creek, FSA Campy and others supply reliable bearings in a variety of configurations to match the bearing seats.
There is nothing inherently wrong with this type of headset. Their biggest weakness is the lack of standardization.