| 2025 | ACL | Sparse Logit Sampling: Accelerating Knowledge Distillation in LLMs. | Anshumann, Mohd Abbas Zaidi, Akhil Kedia, Jinwoo Ahn, Taehwak Kwon, Kangwook Lee, Haejun Lee, Joohyung Lee |
| 2024 | ICML | Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models. | Akhil Kedia, Mohd Abbas Zaidi, Sushil Khyalia, Jungho Jung, Harshith Goka, Haejun Lee |
| 2022 | EMNLP | FiE: Building a Global Probability Space by Leveraging Early Fusion in Encoder for Open-Domain Question Answering. | Akhil Kedia, Mohd Abbas Zaidi, Haejun Lee |
| 2022 | EMNLP | You Only Need One Model for Open-domain Question Answering. | Haejun Lee, Akhil Kedia, Jongwon Lee, Ashwin Paranjape, Christopher D. Manning, Kyoung-Gu Woo |
| 2021 | ACL | Learning to Generate Questions by Learning to Recover Answer-containing Sentences. | Seohyun Back, Akhil Kedia, Sai Chetan Chinthakindi, Haejun Lee, Jaegul Choo |
| 2021 | EACL | Keep Learning: Self-supervised Meta-learning for Learning from Inference. | Akhil Kedia, Sai Chetan Chinthakindi |
| 2021 | EMNLP | Beyond Reptile: Meta-Learned Dot-Product Maximization between Gradients for Improved Single-Task Regularization. | Akhil Kedia, Sai Chetan Chinthakindi, Wonho Ryu |
| 2020 | ICLR | NeurQuRI: Neural Question Requirement Inspector for Answerability Prediction in Machine Reading Comprehension. | Seohyun Back, Sai Chetan Chinthakindi, Akhil Kedia, Haejun Lee, Jaegul Choo |