GPT-4 to LLaMA2 on AWS migration for Enterprise SaaS Analytics

Business Problem

Lyzr is an Antler backed startup, which builds building blocks & Generative AI powered products. NeoAnalyst, one solution from Lyzr, is an AI data analyst platform, which enables business leaders to interact with data through natural language queries. The existing solution was built on GPT-4. The target audience for the platform being enterprise business users, they faced multiple challenges in enterprise deployments, as they scaled:

• Enterprise customer preferences for an open LLM model over OpenAI’s GPT series
• Ensure the entire platform remains compliant with.
• GDPR regulations, ensuring user data privacy and protection.
• SOC2 standards, guaranteeing security, availability, processing integrity, confidentiality, and privacy of customer data.

Additionally, the NeoAnalyst product required the implementation of an agent-based model for facilitating interactions between multiple agents that collectively provide AI data analyst functionality.

About Lyzr.ai

Lyzr is an enterprise Generative AI company that offers private and secure AI Agent SDKs and an AI Management System. Lyzr helps enterprises build, launch and manage secure GenAI applications, in their AWS cloud or on-prem infra. No more sharing sensitive data with SaaS platforms or GenAI wrappers. And no more reliability and integration issues of open-source tools.

Visit Website

Explore Now

Solution

Post evaluation by the GoML Data Scientists, LLaMA2 was the target LLM for migration, which met the above-mentioned criteria. The migration process involved transitioning the existing chatbot system from using ChatGPT’s capabilities by leveraging the enhanced features of LLaMA2.

The overall scope identified by the goML’s GenAI team was broken into 3 parts:

Migrating to LLaMA2

The migration of core components from GPT-4 to a hosted version of LLaMA2 involved transitioning the existing chatbot system to leverage LLaMA2’s enhanced features. This was executed seamlessly by hosting the new system natively on AWS with a serverless architecture, utilizing Lambda-based microservices for improved efficiency and scalability.

Fine-tuning LLaMA2-7B

The 7B version of the LLaMA2 model underwent instruct tuning to enhance its capabilities, ensuring it meets or exceeds the performance of existing OpenAI models. This process involved refining the model for accuracy and high-quality output tailored to the client’s needs.

Optimizing LLaMA2 Performance

The LLaMA2 model was fine-tuned to ensure its output accuracy and performance matched or exceeded that of existing OpenAI models. This optimization guaranteed high-quality, reliable responses from the chatbot system.
There was a need for fine-tuning the LLaMA2-7b chat open-source model & the entire platform was designed to be hosted natively on AWS using a serverless architecture, specifically employing Lambda-based microservices.

GPT-4 to LLaMA2 on AWS migration for Enterprise SaaS Analytics - Lyzr.ai — Click to View in Full Size

Architecture

User Interface and Routing: Frontend application for user interaction, managed by Amazon Route 53 for reliable DNS routing.
API Gateway and Load Balancer: Amazon API Gateway processes API requests, and Elastic Load Balancer distributes traffic across multiple EC2 instances for high availability.
Compute and Serverless Processing: Amazon EC2 instances and AWS Fargate host backend applications, while AWS Lambda handles serverless function execution.
Data Storage and Management: Amazon S3 for object storage, Amazon RDS for relational databases, Amazon DynamoDB for NoSQL storage, and Amazon Redshift for data warehousing.
Search and Analytics: Amazon Elasticsearch Service provides real-time search and analytics capabilities.
Machine Learning and Data Integration: Amazon SageMaker for machine learning model development and deployment, and AWS Glue for data integration and preparation.
Security and Monitoring: AWS IAM for secure access management, AWS CloudWatch for monitoring and observability, and AWS WAF for web application security.