Yi také hraje za čínský národní tým, reprezentoval svou zemi na olympiádě v letech 2004, 2008, 2012.
a 2016, jakož i mistrovství světa FIBA v letech 2006 a 2010. Yi hrál na olympiádě v roce 2004 ve věku 19 let.