Seungjun Lee, KAIST