Yi juga bermain untuk tim nasional China, setelah mewakili negaranya di Olimpiade pada tahun 2004, 2008, 2012 dan 2016.
, serta Kejuaraan Dunia FIBA 2006 dan 2010. Yi bermain di Olimpiade 2004 pada usia 19 tahun.