Cavity model is one of the key computational models for power distribution network simulation. However, the conventional cavity model assumes uniform current distribution along the via and its accuracy is limited to the coupling of close-placed vias. This paper presents an improved cavity model based with gradient current distribution assumption. The loop inductance and surface current distribution are simulated by the proposed cavity model. The comparison between improved and original models is employed to demonstrate the advantage of proposed model.