The present paper deals with a homogenization technique based on the Transformation Field Analysis (TFA) for the study of heterogeneous composite media characterized by nonlinear response. According to the TFA, the behavior of the representative volume element (RVE) is studied accounting for the nonlinear effects by means of the presence of a uniform inelastic strain distribution in the nonlinear constituent of the heterogeneous material. In order to improve the TFA, the assumption of uniformity of the inelastic strain distribution is removed, so that a nonuniform inelastic strain field, better representing the inelasticity distribution in the composite, is considered. In particular, the inelastic strain is represented as a piecewise linear...