I actually like the restricted initiative die in BG3, because you can actually plan a build around going first. Compensating heavy armor with more tankiness instead of less would make for an interesting tradeoff.