保留正文与补采附件。源页面当前未重新核验；具体限制见交付清单。

Revisit consent button We value your privacy We use cookies to enhance your browsing experience, serve personalised ads or content, and analyse our traffic. By clicking "Accept All", you consent to our use of cookies. Customise Reject All Accept All Customise Consent Preferences We use cookies to help you navigate efficiently and perform certain functions. You will find detailed information about all cookies under each consent category below. The cookies that are categorised as "Necessary" are stored on your browser as they are essential for enabling the basic functionalities of the site. ... Show more NecessaryAlways Active Necessary cookies are required to enable the basic features of this site, such as providing secure log-in or adjusting your consent preferences. These cookies do not store any personally identifiable data.

Functional Functional cookies help perform certain functionalities like sharing the content of the website on social media platforms, collecting feedback, and other third-party features.

Analytics Analytical cookies are used to understand how visitors interact with the website. These cookies help provide information on metrics such as the number of visitors, bounce rate, traffic source, etc.

Performance Performance cookies are used to understand and analyse the key performance indexes of the website which helps in delivering a better user experience for the visitors.

Advertisement Advertisement cookies are used to provide visitors with customised advertisements based on the pages you visited previously and to analyse the effectiveness of the ad campaigns.

Uncategorised Other uncategorised cookies are those that are being analysed and have not been classified into a category as yet. No cookies to display. Reject All Save My Preferences Accept All Toggle SearchToggle Mobile Navigation Publish AI, ML & data-science insights to a global community of data professionals. Sign inSubmit an Article LatestEditor's PicksNewsletterShipAI

Write For TDS LinkedInXToggle Search Artificial Intelligence

Managing the Technical Debts of Machine Learning Systems

Explore the practices with implementation codes for sustainably mitigating the cost of speedy delivery John Leung September 26, 202310 min read

Managing the Technical Debt of Machine Learning Systems

Explore the practices (design patterns, version control, and monitoring systems) for sustainably mitigating the cost of speedy delivery—with implementation codes

As the machine learning (ML) community advances over the years, the resources available for developing ML projects are plentiful. For example, we can rely on the generic Python package scikit-learn, which is built on NumPy, SciPy, and matplotlib, for data preprocessing and basic predictive tasks. Or we can leverage the open-source collection of pre-trained models from Hugging Face for analyzing diverse types of datasets. These empower current data scientists to quickly and effortlessly tackle standard ML tasks while achieving moderately good model performance. However, the abundance of ML tools often leads business stakeholders and even practitioners to underestimate the effort required to build enterprise-level ML systems. Particularly when faced with tight project deadlines, the teams may expedite deploying systems to production without giving sufficient technical considerations. Consequently, the ML system often does not address the business needs in a technically sustainable and maintainable manner.

As the system evolves and deploys over time, technical debts accumulate - The longer the implied cost remains unaddressed, the more costly it becomes to rectify them. Photo by Andrea De Santis on Unsplash Photo by Andrea De Santis on Unsplash There are multiple sources of technical debts in the ML system. Some are included below.

1 Inflexible code design to cater to unforeseen requirements

To validate if ML can address the enterprise challenges at hand, many ML projects commence with a proof of concept (PoC). We initially created a Jupyter Notebook or Google Colab environment to explore data, then developed several ad-hoc functions, and created the illusion of nearing project completion for stakeholders. Such systems building directly from PoC may end up consisting mostly ofglue code - the supporting code that connects specific incompatible components but itself does not have the functionality of data analysis. They can be spaghetti-like, hard to maintain, and prone to errors. Photo by Jakob Owens on Unsplash Photo by Jakob Owens on Unsplash Business stakeholders often present alternate requirements from time to time or wish to scale up the project, such as trying out new data sources or new algorithms. We thus often find ourselves frequently revisiting the codebase covering the current preprocessing pipelines and model development processes. Inflexible code design can consequently lead to difficulties in reacting to new changes, or rewrite most of the codes for even minor adjustments.

2 Messiness in the configurations of the ML system

Software engineering programming automates tasks by defining rules for computers to follow, ensuring precise output for the same input. Software engineers are also concerned with the correctness of every corner case. On the other hand, ML systems automate tasks by collecting and feeding feature data into models to achieve desired target results. This experimental process embraces uncertainty and variability. As ML systems mature, they often contain multiple versions of configuration options , such as datasets with different feature combinations, and algorithm-specific learning settings. Photo by Ricardo Viana on Unsplash Photo by Ricardo Viana on Unsplash Input features in ML systems are inherently interconnected. Consider a scenario where feature A is no longer available as an input for your ML systems in production, you are required to re-evaluate the weighting of the remaining features for production serving. However, after 2 months, feature A becomes available again. If you did not systematically save or even mistakenly modify the original configurations, rectifying the decrease in performance would require additional computational resources and time effort.

3 Limited ability to adapt to the evolving external world

ML systems often have dependencies on the external world, and various hidden factors continually evolve but are not being appropriately considered and monitored. External factors leading to potential degradation of model performance image by author External factors leading to potential degradation of model performance (image by author)

In the following sections, we will delve into some great practices for building ML systems and provide illustrative Python code examples to demonstrate their implementation. Imagine that you would like to build a robust traffic system for the city prepared for traffic peaks, so you collected traffic data from sensors over the past two years. Your goal is to predict the traffic patterns (i.e. the no. of vehicles) for the upcoming six months.

Photo by Joey Kyber on Unsplash Photo by Joey Kyber on Unsplash

1 Use design patterns for the ML codebase

To make the code design more flexible and reusable for future requirements, we can leverage design patterns. These patterns serve as the templates for solving common problems in various situations, enabling us to decouple different parts of the codebase. As a result, this helps to improve the comprehension of the codebase and build a common language to communicate quickly about solutions. The two primary components in ML projects are data and algorithms, which can benefit from design patterns.

This creational pattern provides a layer of abstraction for generating objects at runtime. In ML systems, we can implement this pattern by creating a data loader class (CSVDataLoader in this example) that handles the loading, saving, and returning of training/ test data with a consistent data type. We can then declare the DataProcessor interface without specifying the concrete implementation. python

# CSV Data loader class
class CSVDataLoader:
  def __init__(self, file_path):
    self.file_path = file_path

  def get_data(self):
    return pd.read_csv(self.file_path)

  def save_data(self, df):
    df.to_csv(f'data/transformed_{self.file_path}', index=False)

# Interface
class DataProcessor:
  def __init__(self, train_data_loader, test_data_loader):
    self.train_data_loader = train_data_loader
    self.test_data_loader = test_data_loader

  def run(self):
    # Load training and test data using data loaders
    train_df = self.train_data_loader.get_data()
    test_df = self.test_data_loader.get_data()

# Create a data processor instance
process = DataProcessor(
 train_data_loader=CSVDataLoader(file_path='train.csv'),
 test_data_loader=CSVDataLoader(file_path='test.csv')
)

# Run the data processing pipeline
process.run()

This approach allows you to extend the code without having to re-implement the DataProcessor. For example, if you want to load the dataset from a JSON file, you can simply create a new class JSONDataLoader as an instance of the data loader for declaration.

Since there is no one-size-fits-all algorithm for ML problems with varying data distribution, we often find ourselves switching and experimenting between algorithms during prototyping or project enhancement. We can apply the Strategy Pattern by creating a new class DataTransformer for feature engineering and another class LGBMModel for encapsulating the strategy of fitting and predicting using the LightGBM model. python

class DataTransformer:
  def transform_data(self, train_df, test_df):
    for idx, df in enumerate([train_df, test_df]):
      df['DateTime'] = pd.to_datetime(df['DateTime'])

      # Build 'Time' column
      df['Time'] = [date.hour * 3600 + date.minute * 60 + date.second for date in df['DateTime']]

      # Convert DateTime to Unix timestamp
      unixtime = [time.mktime(date.timetuple()) for date in df['DateTime']]
      df['DateTime'] = unixtime

      # Perform one-hot encoding on the DataFrame
      df = pd.get_dummies(df)

      if idx == 0:
        # Split training DataFrame into features (X_train) and target (y_train)
        X_train = df.drop(['Vehicles'], axis=1)
        y_train = df[['Vehicles']]
      elif idx == 1:
        # Store test DataFrame
        X_test = df

    return X_train, y_train, X_test

ruby

class LGBMModel:
  def __init__(self, num_leaves, n_estimators):
    self.model = lgb.LGBMRegressor(
      num_leaves=num_leaves,
      n_estimators=n_estimators
    )

  def fit(self, X, y):
    self.model.fit(X, y)
    self.model.booster_.save_model('model/lgbm_model.txt')
    return self

  def predict(self, X):
    predictions = self.model.predict(X)
    return predictions

The implementation and declaration of the interface DataProcessor are provided below. This is the end-to-end process that includes loading training and test data using train_data_loader and test_data_loaderrespectively, transforming the data using data_transformer, and fitting the model to the transformed data using model. As a result, we can predict the number of vehicles in each record of the test dataset. python

# Interface
class DataProcessor:
  def __init__(self, train_data_loader, test_data_loader, data_transformer, model):
    self.train_data_loader = train_data_loader
    self.test_data_loader = test_data_loader
    self.data_transformer = data_transformer
    self.model = model

  def run(self):
    # Load train and test data using data loaders
    train_df = self.train_data_loader.get_data()
    test_df = self.test_data_loader.get_data()

    # Transform the data using the data transformer
    X_train, y_train, X_test = self.data_transformer.transform_data(train_df, test_df)

    # Fit the model and make prediction
    self.model.fit(X_train, y_train)
    test_df['Vehicles'] = self.model.predict(X_test)

    # Save the transformed training data and test data
    self.train_data_loader.save_data(pd.concat([X_train, y_train], axis=1))
    self.test_data_loader.save_data(test_df)

# Create a data processor instance
process = DataProcessor(
    train_data_loader=CSVDataLoader(file_path='train.csv'),
    test_data_loader=CSVDataLoader(file_path='test.csv'),
    data_transformer=DataTransformer(),
    model=LGBMModel(num_leaves=16, n_estimators=80)
)

# Run the data processing pipeline
process.run()

You can easily add new blocks of code to implement other data transformation ideas or algorithms. Similar to the Factory Pattern, these changes would not require you to modify the interface DataProcessor. This design makes it easier to maintain the code, even if you have a long list of algorithms. The behavior of the ML system can dynamically vary based on the chosen strategy. Of course, the above code implementation is only the preliminary template for development. For example, we can further enhance the code by covering data validation, implementing a hyperparameter tuning mechanism, and evaluating the model.

2 Version control of the ML systems

Throughout the complex process of model development and management, we require proper version control. This enables us to maintain a history of modifications done by ourselves or team members and track the version in the local environment relative to the components of the ML system, including data, trained models, and hyperparameters. We can thus address some common questions, including:

Here we demonstrate how to utilize the versioning features in DVC, which works best in the Git repository for tracking our original traffic data, transformed traffic data, and LGBM models. shell

# Initialise a Git and DVC project in the current working directory
git init
dvc init

# Capture the current state of transformed data in folder 'data' and latest LGBM model in folder 'model'
dvc add data model

# Commit the current state of 1st version
git add data.dvc model.dvc .gitignore
git commit -m "First LGBM model, with Time feature"
git tag -a "v1.0" -m "model v1.0, Time feature"

Let's consider a scenario where we have made the following changes in the second version of data processing:

python

df['Weekday'] = [datetime.weekday(date) for date in df.DateTime]

python

model=LGBMModel(num_leaves=20, n_estimators=90)

With the following commands, we can track the second version of the data and model in DVC and commit the .dvc files that point to them with Git. shell

git add data.dvc model.dvc
git commit -m "Second LGBM model, with Time and Weekday features"
git tag -a "v2.0" -m "model v2.0, Time and Weakday features"

Though the workspace is currently locating the second version of our data and model, we can easily switch back and restore to the first snapshot whenever necessary. shell

git checkout v1.0
dvc checkout

The above commands cover the basic operations. We can further leverage the tool for project organization and collaboration. Examples of use cases are to understand how the data was built initially and compare model metrics among experiments.

3 Test and monitor the ML systems continuously

Once the enhanced model can generate predictions, it is essential to perform sanity checks before releasing it to production. This is achieved by fitting a random sample set of online data into the latest model offline and examining the results from various perspectives.

Even after the latest version of the ML system is released, ongoing monitoring is necessary to account for evolving external environments.

Wrapping it up

We have explored several effective practices that can be implemented to tackle technical debts in developing and deploying ML systems.

Before you go

If you enjoy this reading, I invite you to **** follow my Medium page and Linkedin page. By doing so, you can stay updated with exciting content related to data science side projects, Machine Learning Operations (MLOps) demonstrations, and project management methodologies.

Monitoring Machine Learning Models in Production: Why and How? Optimizing Your Strategies with Approaches Beyond A/B Testing

Register Now to Continue Reading

Unlock Towards Data Science with a Free Account

All articles on Towards Data Science are available to registered users. Create a free account today to access this article and thousands more. Register for freeSign in to your account

Written By

John Leung

See all from John Leung Artificial IntelligenceMachine LearningPythonTechnical DebtVersion Control Share This Article

Towards Data Science is a community publication. Submit your insights to reach our global audience and earn through the TDS Author Payment Program. Write for TDS

Related Articles

Exploring Google Analytics Realtime Data with Python

Programming Take full advantage of all the Google Analytics features and data using its REST API and Python Martin Heinz November 29, 20218 min read

Top 7 Machine Learning resources I wish I knew earlier

Machine Learning Not teaching materials, just useful websites Dmytro Nikolaiev (Dimid) April 30, 20228 min read

Machine Learning in Breast Cancer Research

Machine Learning How explainable machine learning can be applied for pattern recognition in breast cancer research Jarrett Evans August 18, 20226 min read

Execution Times in Python

Measure the execution time of your code in Python the right way Diego Barba June 22, 20226 min read

Airflow Data Intervals: A Deep Dive

Machine Learning Building idempotent and re-playable data pipelines Giorgos Myrianthous January 7, 20259 min read

6 Fundamental Visualizations for Data Analysis

Artificial Intelligence A practical guide with Matplotlib Soner Yıldırım December 4, 20206 min read

A Practical Introduction to LLMs

Artificial Intelligence 3 levels of using LLMs in practice Shaw Talebi July 13, 20237 min read

Tuning with HDBSCAN

Machine Learning A Quick Example of How to Tune Density Based Clustering from the Trenches Clustering is... Charles Frenzel September 2, 20217 min read

Understand Bayes Rule, Likelihood, Prior and Posterior

Machine Learning A case study based introduction to using Bayes rule and how it compares with a frequentist, pessimistic and optimistic approaches to... Ahmar Shah, PhD (Oxford) December 25, 20209 min read

Your home for data science and AI. The world's leading publication for data science, data analytics, data engineering, machine learning, and artificial intelligence professionals. Subscribe to Our NewsletterWrite For TDS•About•Advertise•Privacy Policy•Terms of Use © Insight Media Group, LLC 2026•The New Stack•roadmap.sh•Creator Network

Recommended Articles

Close