Parameter Efficient Finetuning for LLM

AWS Sagemaker Canvas Enterprise Project Roadmap

Project Overview

Advanced research and development project focused on optimizing Parameter Efficient Fine-tuning (PEFT) methods for enterprise-scale Large Language Model deployment in AWS SageMaker Canvas.

Problem Statement

This research addressed the challenge of determining optimal hyperparameters and training configurations for state-of-the-art (SoTA) Large Language Models when implementing Parameter Efficient Fine-tuning (PEFT) methods. The goal was to establish best practices for fine-tuning these models while maintaining computational efficiency and achieving superior performance across diverse use cases.

Business Impact

Enabled enterprise customers to fine-tune Large Language Models for domain-specific business applications while maintaining optimal cost efficiency and performance. This capability was integrated into AWS SageMaker Canvas, Amazon's managed machine learning service for no-code/low-code model development.

Technical Details & Contributions

  • Designed and executed large-scale experiments to understand the effects of fine-tuning LLMs using PEFT methods (LoRA, qLoRA), uncovering overall performance envelopes and investigating failure conditions.
  • Conducted comprehensive analysis of the hyperparameter landscape for PEFT using multiple methodologies and performance profiling techniques.
  • Analyzed performance trade-offs when applying PEFT to domain-specific datasets in resource-constrained environments.
  • Experimented with multiple distributed training paradigms including Fully Sharded Data Parallel (FSDP), Distributed Data Parallel (DDP), and Data Parallel (DP).

Keywords

  • Large Language Models
  • PEFT
  • LoRA
  • qLoRA
  • AWS SageMaker Canvas
  • Distributed Training

References