Artificial Intelligence (AI) and Machine Learning (ML) are a thing. They are here and here to stay. The NT Concepts Data Science Team is hard at work applying differentiating methods and models to complex data sets for our federal customers. We try our hands at everything from computer vision and natural language processing (NLP) to trickier sensor data-type processing. Our successes are celebratory, but we want to share with you what has tripped us up along the way, how we go about our process, and what kinds of cool problem sets are found in the National Security, Defense, and First Responder spaces. In our introductory post in a series on AI/ML development and application, the NT Concepts Data Science Team guides you through our process using our ML Life Cycle. We offer an expert overview while highlighting the challenges inherent in each step. In subsequent articles, we will more deeply examine each stage of ML—hiccups, lessons learned, successes and all!—with a view towards what to watch out for when moving through the process.

Structuring the machine learning process

Amid testing, fiddling, and a lot of internal R&D-type activities, we tried to pull some threads of continuity through the processes our team was iteratively enacting in pursuit of data science. We took a hard look at our ML, Deep Learning, and Unsupervised Learning techniques, as well as our more traditional statistical approaches, and established a cycle that encompasses all the relevant steps while allowing for modification for non-traditional or specialized applications of data science.

ML Lifecycle
The NT Concepts ML lifecycle

The resulting ML lifecycle facilitates our Agile AI process, generates repeatable processes, and contextualizes the challenges data science teams face when executing AI/ML. Within each quadrant, the tasks present “hidden challenges.” It is critical that the data science team addresses these challenges to ensure the initial “minimum viability” and eventual operations-level accuracy of the models produced in the cycle. The four quadrants are:

  1. Project Design
  2. Data Preparation
  3. Model Fitting
  4. Inference & Deployment

Quadrant One: Project Design

Not surprisingly, the first quadrant focuses on selecting the appropriate methods to address the predictive, analytic, or automation task at hand. It is one of the most difficult phases in the lifecycle and one that requires much collaboration between scientist and decision maker.

Because AI and ML have become buzzwords in the technical community, the biggest challenge in quadrant one comes down to determining whether the problem truly requires an ML solution. It is possible an advanced analytic, statistical, or another approach could suffice. 

As a rule, the best way to determine whether an ML solution is required is to ask decision makers to complete one of the two following sentences:

“We need to automate X because…”
“We need to predict X because…”

The approach to this phase contains four steps:

  1. Desired Output Determination
  2. Target Data Identification
  3. Model Selection
  4. Ontology Engineering

While ontology engineering is arguably the most technically-involved step in the first quadrant, the biggest challenges occur in the process of identifying the target data available and determining the model best suited for the task. Once we’ve done those two things, it is on to the task on which the bulk of our time is spent: data wrangling.

Quadrant Two: Data Preparation

A challenge familiar to any and all involved in data science is pre-processing and labeling data. The process of data wrangling (i.e., moving data to a central repository and transforming it into a schema conducive to ingest into a popular ML framework) ends up requiring much more time than the actual model tuning and testing itself. However, data labeling is the largest challenge as there seems to be no good way to avoid egregious human capital expenses and hours of tedious, error-prone work.

The specific steps within this phase are:

  1. Data Wrangling and Cleaning
  2. Feature Engineering
  3. Data Labeling

The stage that follows data preparation is often the phase people most readily associate with the field of data science. The third stage involves true data scientists (as opposed to data engineers and software developers) as rigorous testing and evaluation must be directed towards model tuning and performance.

Quadrant Three: Model Fitting

Even though the bulk of the work happens during the data preparation and labeling phase (quadrant two), model preparation is the meat and potatoes of ML. This phase includes three steps:

  1. Cloud Architecting
  2. Model Training
  3. Model Evaluation and Validation

The most prominent challenge in this stage is in the precision mathematics (and a whole lot of trial and error) that lead to model validation through hyperparameter tuning and model testing.

Quadrant Four: Inference & Deployment

At this point in our ML Lifecycle, the team is prepared to leverage the model to make viable predictions, post-process the results, and visualize them. This is a joint effort that involves analysts, data scientists, and UI/UX engineers. In some cases, depending on the task, specialized visualization techniques are required such as post-process triangulation algorithms to calculate a more precise location for a predicted object. The three steps are:

  1. Prediction
  2. Post-processing and Visualization
  3. Operational Deployment

Assuming the model is generating accurate results that are visually represented, all that is left is for it to be used for its intended purpose. In the National Security environment, operational deployment of AI algorithms is one of the biggest challenges, both due to compute power constraints of downrange environments and discrepancies between unclassified training data and the mission data on which the model may be expected to perform inference.

Conclusion

Inherent across the ML Lifecycle, additional challenges exist such as model brittleness and bias. In subsequent articles, we will examine and address the challenges hidden within each quadrant in depth, along with some of the intangible influencers (such as bias) that can determine the success of using AI/ML to solve the most challenging questions.