End-to-end JumpStart solution templates - Amazon SageMaker AI

End-to-end JumpStart solution templates

Important

As of November 30, 2023, the previous Amazon SageMaker Studio experience is now named Amazon SageMaker Studio Classic. The following section is specific to using the Studio Classic application. For information about using the updated Studio experience, see Amazon SageMaker Studio.

Note

JumpStart Solutions are only available in Studio Classic.

SageMaker AI JumpStart provides one-click, end-to-end solutions that are designed to address common machine learning use cases. They use proven algorithms for their domains and provide a complete workflow which typically includes data processing, model training, deployment, inference, and monitoring. Explore the following use cases for more information on available solution templates.

Choose the solution template that best fits your use case from the JumpStart landing page. When you choose a solution template, JumpStart opens a new tab showing a description of the solution and a Launch button. When you select Launch, JumpStart creates all of the resources that you need to run the solution, including training and model hosting instances. For more information on launching a JumpStart solution, see Launch a Solution.

After launching the solution, you can explore solution features and any generated artifacts in JumpStart. Use the Launched JumpStart assets menu to find your solution. In your solution's tab, select Open Notebook to use provided notebooks and explore the solution’s features. When artifacts are generated during launch or after running the provided notebooks, they're listed in the Generated Artifacts table. You can delete individual artifacts with the trash icon ( The trash icon for JumpStart. ). You can delete all of the solution’s resources by choosing Delete solution resources.

Demand forecasting

Demand forecasting uses historical time series data in order to make future estimations in relation to customer demand over a specific period and streamline the supply-demand decision-making process across businesses.

Demand forecasting use cases include predicting ticket sales in the transportation industry, stock prices, number of hospital visits, number of customer representatives to hire for multiple locations in the next month, product sales across multiple regions in the next quarter, cloud server usage for the next day for a video streaming service, electricity consumption for multiple regions over the next week, number of IoT devices and sensors such as energy consumption, and more.

Time series data is categorized as univariate and multi-variate. For example, the total electricity consumption for a single household is a univariate time series over a period of time. When multiple univariate time series are stacked on each other, it’s called a multi-variate time series. For example, the total electricity consumption of 10 different (but correlated) households in a single neighborhood make up a multi-variate time series dataset.

Solution name Description Get started
Demand forecasting Demand forecasting for multivariate time series data using three state-of-the-art time series forecasting algorithms: LSTNet, Prophet, and SageMaker AI DeepAR.

GitHub »

Credit rating prediction

Use JumpStart's credit rating prediction solutions to predict corporate credit ratings or to explain credit prediction decisions made by machine learning models. Compared to traditional credit rating modeling methods, machine learning models can automate and improve the accuracy of credit prediction.

Solution name Description Get started
Corporate credit rating prediction Multimodal (long text and tabular) machine learning for quality credit predictions using AWS AutoGluon Tabular. GitHub »
Graph-based credit scoring Predict corporate credit ratings using tabular data and a corporate network by training a Graph Neural Network GraphSAGE and AWS AutoGluon Tabular model. Find in Amazon SageMaker Studio Classic.
Explain credit decisions Predict credit default in credit applications and provide explanations using LightGBM and SHAP (SHapley Additive exPlanations).

GitHub »

Fraud detection

Many businesses lose billions annually to fraud. Machine learning based fraud detection models can help systematically identify likely fraudulent activities from a tremendous amount of data. The following solutions use transaction and user identity datasets to identify fraudulent transactions.

Solution name Description Get started
Detect malicious users and transactions Automatically detect potentially fraudulent activity in transactions using SageMaker AI XGBoost with the over-sampling technique Synthetic Minority Over-sampling (SMOTE).

GitHub »

Fraud detection in financial transactions using deep graph library Detect fraud in financial transactions by training a graph convolutional network with the deep graph library and a SageMaker AI XGBoost model.

GitHub »

Financial payment classification Classify financial payments based on transaction information using SageMaker AI XGBoost. Use this solution template as an intermediate step in fraud detection, personalization, or anomaly detection.

Find in Amazon SageMaker Studio Classic.

Computer vision

With the rise of business use cases such as autonomous vehicles, smart video surveillance, healthcare monitoring and various object counting tasks, fast and accurate object detection systems are rising in demand. These systems involve not only recognizing and classifying every object in an image, but localizing each one by drawing the appropriate bounding box around it. In the last decade, the rapid advances of deep learning techniques greatly accelerated the momentum of object detection.

Solution name Description Get started
Visual product defect detection Identify defective regions in product images either by training an object detection model from scratch or fine-tuning pretrained SageMaker AI models.

GitHub »

Handwriting recognition Recognize handwritten text in images by training an object detection model and handwriting recognition model. Label your own data using SageMaker Ground Truth. GitHub »
Object detection for bird species Identify birds species in a scene using a SageMaker AI object detection model.

Find in Amazon SageMaker Studio Classic.

Extract and analyze data from documents

JumpStart provides solutions for you to uncover valuable insights and connections in business-critical documents. Use cases include text classification, document summarization, handwriting recognition, relationship extraction, question and answering, and filling in missing values in tabular records.

Solution name Description Get started
Privacy for sentiment classification Anonymize text to better preserve user privacy in sentiment classification.

GitHub »

Document understanding Document summarization, entity, and relationship extraction using the transformers library in PyTorch.

GitHub »

Handwriting recognition Recognize handwritten text in images by training an object detection model and handwriting recognition model. Label your own data using SageMaker Ground Truth. GitHub »
Filling in missing values in tabular records Fill missing values in tabular records by training a SageMaker Autopilot model.

GitHub »

Predictive maintenance

Predictive maintenance aims to optimize the balance between corrective and preventative maintenance by facilitating the timely replacement of components. The following solutions use sensor data from industrial assets to predict machine failures, unplanned downtime, and repair costs.

Solution name Description Get started
Predictive maintenance for vehicle fleets Predict vehicle fleet failures using vehicle sensor and maintenance information with a convolutional neural network model.

GitHub »

Predictive maintenance for manufacturing Predict the remaining useful life for each sensor by training a stacked Bidirectional LSTM neural network model using historical sensor readings.

GitHub »

Churn prediction

Customer churn, or rate of attrition, is a costly problem faced by a wide range of companies. In an effort to reduce churn, companies can identify customers that are likely to leave their service in order to focus their efforts on customer retention. Use a JumpStart churn prediction solution to analyze data sources such as user behavior and customer support chat logs to identify customers that are at a high risk of cancelling a subscription or service.

Solution name Description Get started
Churn prediction with text Predict churn using numerical, categorical, and textual features with BERT encoder and RandomForestClassifier.

GitHub »

Churn prediction for mobile phone customers Identify unhappy mobile phone customers using SageMaker AI XGBoost.

Find in Amazon SageMaker Studio Classic.

Personalized recommendations

You can use JumpStart solutions to analyze customer identity graphs or user sessions to better understand and predict customer behavior. Use the following solutions for personalized recommendations to model customer identity across multiple devices, to determine the likelihood of a customer making a purchase, or to create a custom movie recommender based on past customer behavior.

Solution name Description Get started
Entity resolution in identity graphs with deep graph library Perform cross-device entity linking for online advertising by training a graph convolutional network with deep graph library.

GitHub »

Purchase modeling Predict whether a customer will make a purchase by training a SageMaker AI XGBoost model.

GitHub »

Customized recommender system

Train and deploy a custom recommender system that generates movie suggestions for a customer based on past behavior using Neural Collaborative Filtering in SageMaker AI.

Find in Amazon SageMaker Studio Classic.

Reinforcement learning

Reinforcement learning (RL) is a type of learning that is based on interaction with the environment. This type of learning is used by an agent that must learn behavior through trial-and-error interactions with a dynamic environment in which the goal is to maximize the long-term rewards that the agent receives as a result of its actions. Rewards are maximized by trading off exploring actions that have uncertain rewards with exploiting actions that have known rewards.

RL is well-suited for solving large, complex problems, such as supply chain management, HVAC systems, industrial robotics, game artificial intelligence, dialog systems, and autonomous vehicles.

Solution name Description Get started
Reinforcement learning for Battlesnake AI competitions Provide a reinforcement learning workflow for training and inference with the BattleSnake AI competitions.

GitHub »

Distributed reinforcement learning for Procgen challenge Distributed reinforcement learning starter kit for NeurIPS 2020 Procgen Reinforcement learning challenge. GitHub »

Healthcare and life sciences

Clinicians and researchers can use JumpStart solutions to analyze medical imagery, genomic information, and clinical health records.

Solution name Description Get started
Lung cancer survival prediction Predict non-small cell lung cancer patient survival status with 3-dimensional lung computerized tomography (CT) scans, genomic data, and clinical health records using SageMaker AI XGBoost.

GitHub »

Financial pricing

Many businesses dynamically adjust pricing on a regular basis in order to maximize their returns. Use the following JumpStart solutions for price optimization, dynamic pricing, option pricing, or portfolio optimization use cases.

Solution name Description Get started
Price optimization

Estimate price elasticity using Double Machine Learning (ML) for causal inference and the Prophet forecasting procedure. Use these estimates to optimize daily prices.

Find in Amazon SageMaker Studio Classic.

Causal inference

Researchers can use machine learning models such as Bayesian networks to represent causal dependencies and draw causal conclusions based on data. Use the following JumpStart solution to understand the causal relationship between Nitrogen-based fertilizer application and corn crop yields.

Solution name Description Get started
Crop yield counterfactuals

Generate a counterfactual analysis of corn response to nitrogen. This solution learns the crop phenology cycle in its entirety using multi-spectral satellite imagery and ground-level observations.

Find in Amazon SageMaker Studio Classic.