A compressor may include a casing defining a discharge cavity and a seal cavity. A rotary shaft may be disposed in the casing, and a shaft seal assembly may be disposed in the seal cavity and about the rotary shaft. An impeller may be coupled with and configured to be driven by the rotary shaft. A balance piston may be integral with the impeller and may define the discharge cavity and the seal cavity. A balance piston seal may be disposed about the balance piston such that the balance piston seal and the balance piston define a radial clearance therebetween. The radial clearance may be configured to provide fluid communication from the impeller to the discharge cavity. A heat shield may be disposed in the discharge cavity, and may be configured to prevent the conduction of heat from the discharge cavity to the seal cavity via the casing.