Headsets are headsets and as long as the diameter, type (threaded or threadless) and installation method (press-in cups or integrated) are the same and compatible with the frame.

FSA does make a "Pig DH" model that is particularly heavy and rugged as it is intended for downhill MTB use but it still works fine on a road bike if the somewhat clunky appearance isn't an issue.

I would recommend getting a cartridge bearing headset if at all possible as they are easier to install and maintain and tend to be more durable. I've had good luck with both FSA and Cane Creek headsets.