China is starting to introduce an upgraded Infantry Fighting Vehicle (IFV), designated WZ502G. Compared to its predecessor, the original WZ502 or ZBD04, it's armor has been increased. The front of the hull and the turret can withstand hits from 30 mm rounds, the sides 14.5 mm rounds and the rear 7.62mm. It has the option of bolting on additional armor as needed.

The amphibious capability has been reduced with the removal of the rear water jets. The turret includes a thermal imaging gun sight. Armament remains a version of Russia's BMP-3 turret with 100 mm main gun, 30 mm cannon and coaxial 7.62 mm machine gun.