Skip to main content
Partnership | 07 May, 2024

How AIShield SecureAIx Platform - AWS Security Lake Integration Enhance AI Security & Risk Management

Manpreet Dash, Mallikarjun Udanashiv

Integration-of-AIShield-SecureAIx-with-AWS-Security-Lake

Executive Summary

1. Specialized AI Security: AIShield SecureAIx delivers targeted security for AI and ML systems, addressing specific threats like model extraction and data poisoning (for ML) and prompt injection (for GenAI systems).

2. Centralized Security Data Management: Integration with AWS Security Lake centralizes AI security and incident data, enhancing threat detection, reporting, and compliance across environments.

3. Scalable Integration for Enhanced AI Protection: This integration offers a scalable security solution that strengthens incident response and security operations, thereby fortifying the protection of AI workloads, applications, and data.

-

The growth of artificial intelligence (AI) and machine learning (ML) technologies has revolutionized numerous industries by enhancing capabilities and automating complex processes. However, this advancement also introduces new vulnerabilities and security challenges, particularly as AI systems become prime targets for sophisticated cyber-attacks. These threats include model extraction, evasion, data poisoning, and model inference attacks for ML models, as well as prompt injection attacks, insecure output handling, and sensitive information disclosure for Generative AI (GenAI)/Large Language Models (LLMs), all of which can compromise the integrity and confidentiality of AI/ML models and their underlying data.

Given the complexity and scale of data that modern AI systems handle, there is a critical need for robust security mechanisms that can not only detect and mitigate such threats but also manage and analyze security data efficiently. This is where the integration of AI security solutions like the AIShield SecureAIx Platform with comprehensive data management platforms such as AWS Security Lake becomes essential. AWS Security Lake provides a centralized, scalable solution for consolidating security data across various sources, making it an ideal platform to handle the extensive data generated by AI security tools.

AIShield SecureAIx Platform for AI Security

AIShield SecureAIx is an AI Application Security (AppSec) solution that redefines how businesses safeguard their AI assets and infrastructure in the face of emerging security threats. The SecureAIx platform, with its Watchtower, AISpectra, and Guardian modules, consolidates all AI security functionalities for ML models and GenAI systems into a single, enterprise-ready platform encompassing the entire lifecycle of MLOps and LLMOps—from development to validation (including LLM validation) to deployment, operation, and monitoring of AI/ML systems.

Figure 1: SecureAIx Platform for End-To-End Security Coverage of ML & GenAI systems across the workflow
Figure 1: SecureAIx Platform for End-To-End Security Coverage of ML & GenAI systems across the workflow

AIShield AISpectra provides automated hacker-level vulnerability analysis and endpoint protection to harden systems against AI security threats. On the operational front, the AISpectra module (the DAST, IAST, and Penetration Testing module) crafts targeted defenses for AI/ML workloads—Threat-Informed Endpoint Defense (EDR) that can be finely tuned and tested directly within the platform for immediate deployment as an AI Firewall to identify and thwart malicious inputs in real-time.

Real-time intrusion detection is the core capability of the Guardian module—the module focused on security in AI/ML Runtime and Usage/Access. Additionally, for GenAI systems, Guardian offers customizable, enterprise-ready guardrails and mitigates security risks of LLMs, referenced in OWASP Top 10 risks for LLMs, such as prompt injection, insecure output handling, and sensitive information disclosure. In a nutshell, the Guardian module enhances real-time endpoint protection of ML and GenAI systems by reporting attack telemetry data to Security Information and Event Management (SIEM) tools and security data lakes such as Amazon Security Lake.

Key Features of AIShield SecureAIx Platform:

  • Vulnerability Scanning (AISpectra Module): Comprehensive AI security risk analysis of AI/ML models.
  • Endpoint Protection (AISpectra Module): Proactive generation of threat-informed defense models (EDR) with attack data for robust model hardening for ML models.
  • Intrusion Detection (Guardian Module): Real-time monitoring and prevention of attacks in GenAI and ML systems, ensuring security in cloud and device-based deployments.
  • Threat Intelligence/Telemetry SIEM (Guardian Module): Enhanced incident alerting and active threat hunting capabilities.
  • Comprehensive Reporting: Instant insights into your AI security posture throughout the AI lifecycle with detailed dashboards for organizational leaders.

AWS Security Lake

AWS Security Lake, a managed security service provided by Amazon Web Services (AWS), streamlines the consolidation of security data from diverse sources into a centralized data lake. Designed to enhance the aggregation, management, and analysis of security data, this service is optimized for both AWS environments and various other security tools. Its key features include data normalization, scalability, and comprehensive integration with analytical tools, which together ensure efficient management of large volumes of security data and compliance with regulatory requirements.

Figure 2: AWS Security Lake automatically centralizes security data from multiple sources, including Open Cybersecurity Schema Framework (OCSF) compliant security events from custom sources.
Figure 2: AWS Security Lake automatically centralizes security data from multiple sources, including Open Cybersecurity Schema Framework (OCSF) compliant security events from custom sources.

The service supports a variety of critical use cases:

  • Comprehensive Security Analysis: It allows organizations to analyze years of security data, identifying trends and potential threats.
  • Centralized Data Management: AWS Security Lake centralizes massive data volumes from multiple sources (including custom sources) into Amazon S3, simplifying access and analysis.
  • Enhanced Compliance Monitoring: It provides a centralized platform for overseeing and reporting on regulatory compliance across various frameworks.
  • Improved Incident Investigation: The service offers increased visibility into security data, enabling organizations to conduct thorough investigations and respond quickly to incidents.
  • Unified Security Across Environments: It ensures seamless security data management across cloud and on-premises setups, enhancing the effectiveness of security operations teams.

AWS Security Lake is especially beneficial for large enterprises or any organization that needs to manage extensive security data across multiple environments and platforms, improving their ability to detect, investigate, and respond to security incidents efficiently.

Integration of AIShield Guardian with Amazon Security Lake

The SecureAIx platform-generated defense model integrates with Amazon Security Lake as a custom source to deliver enhanced real-time monitoring capabilities to security teams for their AI/ML assets, giving them insights into AI security incidents. The security alerts generated by AIShield Threat-Informed Endpoint Defense models (generated by the AISpectra module for ML models' endpoint protection) are OCSF compliant and can be sent to the AWS Security Lake. The defense model supports OCSF-formatted security finding events stored in S3 as a custom source for Amazon Security Lake.

For GenAI systems, the Guardian module mitigates key LLM risks by monitoring inputs (user prompts) and outputs (LLM responses) and enforcing organizational LLM policies. All telemetry data from the LLM guardrails (of the Guardian module) can be supplied to AWS Security Lake. Learn more about the technical details of the integration onhttps://docs.boschaishield.com/amazon-security-lake.

With the AIShield Guardian and AWS Security Lake integration, organizations can harness the strengths of both platforms to create a fortified security environment for their AI assets. The AIShield SecureAIx platform, with its advanced capabilities in vulnerability scanning, endpoint protection, and real-time intrusion detection, offers proactive and reactive security measures tailored specifically for AI and ML models, as well as real-time endpoint defense with Guardian. The Guardian module integrates seamlessly with AWS Security Lake. The real-time telemetry data from the targeted defense models (for ML models) and from the guardrails for GenAI/LLM models can be sent and visualized on AWS Security Lake, facilitating smoother and more efficient security operations.

AIShield is one of the source partners for AWS Security Lake and can send logs and security events to the customer’s security data lake in the OCSF format. This integration enables seamless aggregation of security insights from AIShield into AWS Security Lake, allowing security teams to conduct comprehensive analyses, monitor and report compliance, and respond swiftly and efficiently to detected threats with advanced security operations. The centralized framework of AWS Security Lake ensures that all security data is normalized and readily accessible, enhancing the effectiveness of threat detection and incident response strategies. When Guardian’s capabilities are combined with the centralized data management and analytical prowess of AWS Security Lake, organizations gain unprecedented visibility and control over their security and risk landscape, improving the protection of their AI workloads, applications, and data.

Conclusion

This collaboration not only streamlines the management of security data across hybrid environments but also enhances the capability to effectively combat emerging AI-specific threats, allowing our customers to reliably scale their security operations that includes AI security. For enterprises aiming to safeguard their AI investments while ensuring compliance and operational efficiency, this integrated solution provides a robust, scalable, and highly effective security infrastructure. AIShield has achieved the Advanced Sagemaker Ready status and the AWS Generative AI Competency and is a contributor to the AWS Generative AI Center of Excellence. As AI continues to permeate various sectors, the importance of such comprehensive security solutions cannot be overstated, ensuring that organizations can continue to innovate safely in the AI space.