Unlocking Patient Population Insights Using Smart Subgroups and LLMs

aetion · Healthcare · 2025 · source Agentic: no · promotional bias: high · source-backed · extracted with prompt 0.5.6 (schema v2)

Generated from extraction.json — do not edit by hand. Every claim cites verbatim source evidence; stated means the quote directly asserts the claim, implied means it supports it indirectly.

Problem

Domain: Healthcare real-world evidence analytics

Aetion needed to make patient-population subgroup analysis accessible through natural language for researchers who lacked deep familiarity with the underlying real-world datasets and structured query formats.

Constraints:

Architecture

Aetion added a Smart Subgroups Interpreter to its Discover product by combining traditional unsupervised ML clustering with a Bedrock-hosted LLM interface. AEP computes more than 1,000 AML features per patient, stores feature data in S3 and Aurora, runs on-demand Kubernetes clustering jobs, identifies distinctive cluster features and prevalences, and passes expanded feature descriptions plus statistics into Claude 3 Haiku prompts to name clusters and answer subgroup questions.

Diagram confidence: reconstructed (reconstructed from prose).

flowchart LR
  aetion-discover-user-interface["Aetion Discover user interface"] --> aetion-evidence-platform["Aetion Evidence Platform"]
  aetion-discover-user-interface --> smart-subgroups-interpreter["Smart Subgroups Interpreter"]
  aetion-evidence-platform --> amazon-s3-feature-store["Amazon S3 feature store"]
  aetion-evidence-platform --> smart-subgroups-clustering-job["Smart Subgroups clustering job"]
  amazon-s3-feature-store --> smart-subgroups-clustering-job
  smart-subgroups-clustering-job --> smart-subgroups-interpreter
  smart-subgroups-interpreter --> amazon-aurora["Amazon Aurora"]
  smart-subgroups-interpreter --> prompt-construction-layer["Prompt construction layer"]
  smart-subgroups-interpreter --> amazon-bedrock["Amazon Bedrock"]
  prompt-construction-layer --> claude-3-haiku["Claude 3 Haiku"]
  claude-3-haiku --> amazon-bedrock
Drag to pan · scroll or use +/− to zoom · reconstructed from the source's described component topology

Components

Component Role Interacts with
Aetion Discover user interface Lets healthcare researchers define populations and ask natural-language subgroup questions. Aetion Evidence Platform, Smart Subgroups Interpreter
Aetion Evidence Platform Computes standardized patient features from real-world data. Aetion Discover user interface, Amazon S3 feature store, Smart Subgroups clustering job
Amazon S3 feature store Stores computed patient features with KMS encryption. Aetion Evidence Platform, Smart Subgroups clustering job
Amazon Aurora Provides persistent application and exchange data for the feature. Smart Subgroups Interpreter
Smart Subgroups clustering job Runs topic modeling and classification to assign patients to clusters and identify distinctive features. Amazon S3 feature store, Smart Subgroups Interpreter
Prompt construction layer Combines scientific feature descriptions and prevalence statistics into versioned prompts. Smart Subgroups Interpreter, Claude 3 Haiku
Claude 3 Haiku Generates cluster names and natural-language answers about patient subgroups. Prompt construction layer, Amazon Bedrock
Amazon Bedrock Provides managed access to the selected LLM and security posture. Claude 3 Haiku, Smart Subgroups Interpreter
Smart Subgroups Interpreter Coordinates clustering outputs, prompts, model calls, and user-facing explanations. Aetion Discover user interface, Amazon Aurora, Smart Subgroups clustering job, Prompt construction layer, Amazon Bedrock

Data flow

A user defines a patient population with the Aetion Measure Library. AEP computes standardized features, stores them in S3, and Smart Subgroups trains a topic model and classifier to find clusters and distinctive features. The interpreter expands coded features into human-readable clinical descriptions, injects prevalence statistics into versioned prompt templates, calls Claude 3 Haiku through Bedrock, and returns natural-language cluster names or subgroup answers.

3 supporting quotes
“The workflow begins with users defining a patient population using the Aetion Measure Library (AML), which standardizes variable definitions using scientifically validated algorithms.”summary, stated
“The Smart Subgroups component trains a topic model using these patient features to determine optimal cluster numbers and assign patients to clusters.”summary, stated
“This involves versioned prompt templates that combine multiple information sources”summary, stated

Why this architecture

Aetion used traditional ML to perform the patient clustering and distinctive-feature identification, then used an LLM as an interpretation layer. That division kept quantitative subgroup discovery grounded in computed features while letting non-specialists query and understand the resulting clusters in natural language.

“The combination of unsupervised learning (clustering) with LLMs for interpretation represents a hybrid approach where traditional ML handles the analytical heavy lifting while the LLM serves as an interface layer.”summary, stated
“This division of labor may provide more reliable results than asking an LLM to perform the clustering analysis directly.”summary, stated

Models used: Anthropic Claude 3 Haiku · Topic model · Classification model

Scale

Alternatives considered (roads not taken)

Ask the LLM to perform clustering analysis directly

Rejected because: Aetion used unsupervised ML for clustering and reserved the LLM for interpretation, which may be more reliable.

“traditional ML handles the analytical heavy lifting while the LLM serves as an interface layer.”summary, stated
“This division of labor may provide more reliable results than asking an LLM to perform the clustering analysis directly.”summary, stated

Use a more capable but slower model

Rejected because: Aetion selected Claude 3 Haiku to balance speed with sufficient intelligence for interactive subgroup interpretation.

“they chose Anthropic's Claude 3 Haiku specifically for its speed and what they describe as a "satisfactory intelligence level" for their use case.”summary, stated
“This choice reflects a pragmatic production consideration: balancing model capability with latency requirements for interactive user experiences.”summary, stated

Deploy generative AI as a standalone product

Rejected because: Aetion embedded the feature into its existing Evidence Platform workflow so users could move from exploration to causal inference.

“The LLM capability is embedded within Aetion's broader Evidence Platform rather than deployed as a standalone application.”summary, stated
“Users can move from AI-powered exploration in Discover to rigorous causal inference analysis in Aetion Substantiate.”summary, stated

Patterns

context-engineering (prompting)

Aetion constructed prompts from scientific feature expansions and prevalence statistics so the model could explain coded healthcare clusters.

“This approach represents a form of structured data augmentation in prompts, where domain-specific scientific knowledge and computed statistics are systematically injected into the LLM context”summary, stated

prompt-versioning (operations)

The feature used versioned prompt templates as managed artifacts for iteration and rollback.

“The explicit mention of "versioned prompt templates" indicates that Aetion treats prompts as managed artifacts, likely enabling iteration, testing, and rollback capabilities.”summary, stated

model-routing (model operations)

The system assigned analytical clustering to traditional ML and natural-language interpretation to Claude 3 Haiku.

“The Smart Subgroups Interpreter combines unsupervised machine learning with generative AI in a multi-step pipeline deployed on AWS infrastructure.”summary, stated
“traditional ML handles the analytical heavy lifting while the LLM serves as an interface layer.”summary, stated

batch-offline-processing (infrastructure)

The clustering step runs as an on-demand Kubernetes job rather than permanently provisioned compute.

“Running the clustering component as on-demand Kubernetes jobs suggests an event-driven architecture that scales resources based on user requests rather than maintaining constant compute capacity.”summary, stated

privacy-preserving-processing (security)

Healthcare feature data used KMS encryption at rest, TLS in transit, and Bedrock's managed security posture.

3 supporting quotes
“Healthcare data requires stringent security controls. The architecture incorporates:”summary, stated
“• AWS KMS encryption for data at rest in S3”summary, stated
“• TLS 1.2 encryption for data in transit”summary, stated

Tradeoffs

Hybrid ML plus LLM interpretation

Chose Use traditional ML for clustering and Claude for interpretation over Ask the LLM to discover patient clusters directly.

Rationale: Quantitative clustering remains grounded in computed features while the LLM makes results accessible.

“The combination of unsupervised learning (clustering) with LLMs for interpretation represents a hybrid approach where traditional ML handles the analytical heavy lifting while the LLM serves as an interface layer.”summary, stated

Claude 3 Haiku model choice

Chose A faster, satisfactory-intelligence model over A more capable but potentially slower model in the Claude family.

Rationale: Interactive subgroup exploration needed acceptable latency as well as sufficient clinical-language capability.

“The choice of Claude 3 Haiku over more capable models in the Claude family suggests optimization for latency in an interactive use case.”summary, stated

Embedded platform integration

Chose Add Smart Subgroups Interpreter inside Aetion's Evidence Platform over Ship a standalone generative AI tool.

Rationale: Embedding AI in the existing workflow let users move from exploratory subgroup discovery to rigorous causal inference.

“This integration pattern, referred to as "AetionAI," represents an approach where generative AI enhances existing workflows rather than replacing them.”summary, stated

Gotchas (what bit them)

Healthcare cluster interpretation needs domain context before the LLM can produce meaningful labels.

Root cause: Raw AML feature codes are too terse for reliable natural-language interpretation.

Resolution: Aetion expanded features into human-readable scientific descriptions and injected prevalence statistics into prompts.

“The feature expansion technique—where coded medical features are enriched with detailed descriptions before being sent to the LLM—addresses the challenge of domain-specific terminology”summary, stated
“• Statistical context injection: The prompts include the top feature prevalences for each subgroup. For instance: "In Smart Subgroup 1 the relative prevalence of 'Cornea and external disease (EYE001)' is 30.32%. In Smart Subgroup 1 the relative prevalence of 'Glaucoma (EYE003)' is 9.94%..."”summary, stated

The public case study does not describe evaluation or guardrails for medical accuracy.

Root cause: The blog focuses on architecture and reported benefits rather than LLM quality governance.

Resolution: The extraction should treat reported healthcare outcomes as promotional and note missing validation details.

3 supporting quotes
“The case study does not address several important LLMOps concerns:”summary, stated
“• Evaluation methodology for LLM response quality”summary, stated
“• Guardrails or validation for medical accuracy of generated content”summary, stated

Latency mattered enough to influence model choice.

Root cause: The feature was an interactive user-facing healthcare analytics experience.

Resolution: Aetion chose Claude 3 Haiku for speed while accepting a satisfactory intelligence level for the task.

“This choice reflects a pragmatic production consideration: balancing model capability with latency requirements for interactive user experiences.”summary, stated

Evaluation

The article reports product outcomes and architectural choices but does not disclose a detailed evaluation methodology for LLM response quality, medical accuracy, guardrails, monitoring, or user feedback.

Metrics: Over 1,000 AML features per patient · Days-to-minutes reported time savings · No disclosed LLM response-quality metric · No disclosed medical accuracy guardrail metric

3 supporting quotes
“The case study does not address several important LLMOps concerns:”summary, stated
“• Evaluation methodology for LLM response quality”summary, stated
“These gaps may simply reflect the scope of the blog post rather than actual limitations of the implementation, but they represent areas that production healthcare AI systems typically need to address carefully.”summary, stated

Outcomes

The system handles patient populations described by more than 1,000 AML features per patient and is part of a platform used by biopharma, payor, and regulatory organizations across the US, Canada, Europe, and Japan.

3 supporting quotes
“The system supports several types of natural language queries:”summary, stated
“The LLM responses cite relevant subgroup statistics, grounding the natural language output in the underlying data.”summary, stated
“The claimed benefit is reducing the time to generate decision-grade evidence from days to minutes, without requiring support staff.”summary, stated