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