University of Alberta, Alberta Machine Intelligence Institute (AMII) – Edmonton, Canada
Machine Learning Research Intern (March 2024 - Present)
- ProtoECGNet: Worked under the supervision of Dr. Russ Greiner, Dr. Padma Kaul, and Dr. Sunil Kalmady Vasu. Developed ProtoECGNet, a prototype-based interpretable diagnostic deep learning model for electrocardiograms (ECGs). The model provides low-level statement code predictions along with the final diagnostic labels.
- Explainability: Integrated LLMs with RAG to provide semantic explanations of diagnoses, similar to human cardiovascular experts.
- Impact: Two variations of ProtoECGNet ranked in the top 10 in IEEE Computing in Cardiology Conference.
Information Processing and Transmission Lab – Islamabad, Pakistan
Machine Learning Research Intern (March 2024 - Present)
- Work: Developing domain-specific LLMs for 6G networks, focusing on network modeling, channel state prediction, and dynamic spectrum allocation. Achieved 94.24% optimal spectral efficiency and 86.51% optimal energy efficiency in D2D wireless networks.
- Submitted: The work has been submitted to the IEEE International Conference on Communications (ICC) 2024.
Deep Learning Lab, National Centre of Artificial Intelligence (NCAI) – H-12, Islamabad
Research Intern (June 2023 - May 2024)
- QuickFormer - GPT for Edge Devices: Proposed and implemented a faster transformer architecture named QuickFormer, achieving an 8x speed-up during training and inference, and a 10x reduced memory footprint.
- EEGFormer: Collaborated with a PhD student to design and implement a custom transformer architecture, EEGFormer, for epilepsy detection with 85% accuracy and an F1 score of 0.71.
High Performance Computing Lab, NUST – H-12, Islamabad
Machine Learning Research Intern (June 2022 - September 2022)
- Intelligent Parking Management: Designed and implemented an IoT-enabled parking management framework, reducing average parking search times by 35% compared to the unintelligent algorithm.
Optical Networks and Technologies Lab (ONT), NUST – H-12, Islamabad
Machine Learning Research Intern (March 2023 - January 2024)
- Virtual Network Embedding: Developed a simulation for an integrated network for 6G, with a focus on free-space optical communication and auto-encoder-based signal reconstruction.
- Optimal BBU Placement: Extended the PhD student's work on optimal Baseband Unit (BBU) placement, implementing an algorithm for 5G networks in online traffic scenarios.