IP Library Granted Patent US 12,423,591
Granted Patent B2
US 12,423,591 · App. 17/538,309 · Granted Sep 23, 2025

Annotation of a machine learning pipeline with operational semantics to support distributed lineage tracking

Inventors: Mudhakar Srivatsa (White Plains, NY); Raghu Kiran Ganti (White Plains, NY); Carlos Henrique Andrade Costa (White Plains, NY); Linsong Chu (White Plains, NY); Joshua M. Rosenkranz (White Plains, NY)
Assignee: International Business Machines Corporation
G06N5/02G06F8/75G06F8/77G06F11/3608G06N20/00
View Patent ↗
Loading inventors, assignments & file history…
Monitor This Case
Get email alerts when status or documents change.
Order Certified Copies
Most orders are placed with the USPTO same day — all within 24 business hours.
Order via The Patent Place →
Pre-filled with this patent's details
Quick Facts
Patent No.
US 12,423,591
App. No.
17/538,309
Granted
Sep 23, 2025
Kind
B2
Abstract

A system, computer program product, and method are provided for distributed data workflow semantics. A pipeline, such as a machine learning (ML) pipeline, is represented in a data flow graph (DFG). The represented pipeline is subject to annotations, with the annotations including pipeline nodes and object references. The pre-processed pipeline is subject to execution or processing with the annotated object references capturing object lineage. Output from the executed pipeline is constructed and a corresponding control signal is formatted to dynamically and selectively control an operatively coupled physical hardware device or software.

Claims (60)

1. A computer system comprising:

a processor set:

one or more computer readable storage media; and

program instructions stored on the one or more computer readable storage media to cause the processor set to perform operations comprising:

pre-processing a pipeline configured to train a machine learning (ML) model, the pipeline represented in a data flow graph (DFG), including:

annotating one or more nodes of the DFG with two or more operational semantics for pipeline operations; and

selectively annotating one or more output object references to a corresponding input object and a prior node state for an output object from the DFG;

executing the pipeline represented in the DFG with the selectively annotated one or more output object references, including capturing an object lineage to provide data of how an object in the pipeline is produced with respect to the output object using the selectively annotated one or more output object references;

identifying provenance of one or more objects represented in the pipeline corresponding to a generated output including performance of the executed pipeline using the object lineage;

selectively applying a remediation action to the DFG based on the provenance of the one or more objects corresponding to the generated output; and

restarting the pipeline by executing the pipeline from a location in a sub-graph of the DFG where the remediation action was selectively applied.

2. The computer system of claim 1 , wherein the operations further comprise:

dynamically configuring and issuing a control signal configuration, the control signal configuration based on a constructed output to an operatively coupled physical hardware device, a process controlled by software, or a combination thereof, associated with the DFG, the control signal configuration selectively controls a physical state of the operatively coupled physical hardware device, the software, or a combination thereof.

3. The computer system of claim 2 , wherein the executing of the pipeline represented in the DFG with the one or more output object references comprises:

tracking transformation of data across the pipeline.

4. The computer system of claim 3 , wherein the identifying of the provenance of the one or more objects comprises:

identifying the sub-graph in the DFG and an input object reference that produced the constructed output.

5. The computer system of claim 1 , wherein capturing of the object lineage comprises:

retaining a posterior node state for the one or more output object references from the DFG.

6. The computer system of claim 5 , wherein the operations further comprise:

attaching the object lineage as object metadata.

7. A computer program product comprising:

one or more computer readable storage media; and

program instructions stored on the one or more computer readable storage media to perform operations comprising:

pre-processing a pipeline represented in a data flow graph (DFG), the pipeline configured to train a machine learning (ML) model, including:

annotating one or more nodes of the DFG with two or more operational semantics for pipeline operations; and

selectively annotating one or more output object references to a corresponding input object and a prior node state for an output object from the DFG;

executing the pipeline represented in the DFG with the selectively annotated one or more output object references, including capturing an object lineage to provide data of how an object in the pipeline is produced with respect to the output object using the selectively annotated one or more output object references;

identifying provenance of one or more objects represented in the pipeline corresponding to a generated output including performance of the executed pipeline using the object lineage;

selectively applying a remediation action to the DFG based on the provenance of the one or more objects corresponding to the generated output; and

restarting the pipeline by executing the pipeline from a location in a sub-graph of the DFG where the remediation action was selectively applied.

8. The computer program product of claim 7 , wherein the operations further comprise:

dynamically configuring and issuing a control signal configuration, the control signal configuration based on a constructed output to an operatively coupled physical hardware device, a process controlled by software, or a combination thereof, associated with the DFG, the control signal configuration selectively controls a physical state of the operatively coupled physical hardware device, the software, or a combination thereof.

9. The computer program product of claim 8 , wherein the executing the pipeline represented in the DFG with the one or more output object references comprises:

tracking transformation of data across the pipeline.

10. The computer program product of claim 9 , wherein the identifying the provenance of the one or more objects comprises:

identifying the sub-graph in the DFG and an input object reference that produced the constructed output.

11. The computer program product of claim 7 , wherein the capturing the object lineage comprises:

retaining a posterior node state for the one or more output object references from the DFG.

12. The computer program product of claim 11 , wherein the operations further comprise:

attaching the object lineage as object metadata.

13. A computer implemented method comprising:

pre-processing a pipeline represented in a data flow graph (DFG), the pipeline configured to train a machine learning (ML) model, including:

annotating one or more nodes of the DFG with two or more operational semantics for pipeline operations; and

selectively annotating one or more output object references to a corresponding input object and a prior node state for an output object from the DFG;

executing the pipeline represented in the DFG with the selectively annotated one or more output object references, including capturing an object lineage to provide data of how an object in the pipeline is produced with respect to the output object using the selectively annotated one or more output object references;

identifying provenance of one or more objects represented in the pipeline corresponding to a generated output including performance of the executed pipeline using the object lineage;

selectively applying a remediation action to the DFG based on the provenance of the one or more objects corresponding to the generated output; and

restarting the pipeline by executing the pipeline from a location in a sub-graph of the DFG where the remediation action was selectively applied.

14. The computer implemented method of claim 13 , further comprising:

dynamically configuring and issuing a control signal configuration, the control signal configuration based on a constructed output to an operatively coupled physical hardware device, a process controlled by software, or a combination thereof, associated with the DFG, the control signal configuration selectively controls a physical state of the operatively coupled physical hardware device, the software, or a combination thereof.

15. The computer implemented method of claim 14 , wherein executing the pipeline represented in the DFG with the one or more output object references further comprises:

tracking transformation of data across the pipeline.

16. The computer implemented method of claim 15 , wherein identifying the provenance of the one or more objects further comprises:

identifying the sub-graph in the DFG and an input object reference that produced the constructed output.

17. The computer implemented method of claim 13 , wherein capturing the object lineage further comprises:

retaining a posterior node state for the one or more output object references from the DFG.

18. The computer implemented method of claim 17 , further comprising:

attaching the object lineage as object metadata.

19. The computer implemented method of claim 13 , wherein the two or more operational semantics comprises an input combination, a firing combination, and a state, or a combination thereof.

Assignments (1)
ASSIGNMENT OF ASSIGNOR'S INTEREST Recorded Nov 30, 2021
From: SRIVATSA, MUDHAKAR; GANTI, RAGHU KIRAN; ANDRADE COSTA, CARLOS HENRIQUE; CHU, LINSONG; ROSENKRANZ, JOSHUA M.
To: INTERNATIONAL BUSINESS MACHINES CORPORATION
Reel/Frame 058245/0101 →
Continuity (1)
Related Publication 20230169354A1 · Jun 1, 2023
References Cited (56)
US 6370589B1 · Sieber et al. · 2002 [cited by applicant]
US 8356185B2 · Olson et al. · 2013 [cited by applicant]
US 8537160B2 · Hargrove et al. · 2013 [cited by applicant]
US 8928665B2 · Fonseca Da Trindade · 2015 [cited by applicant]
US 9053437B2 · Adler · 2015 [cited by applicant]
US 9158797B2 · Stanfill · 2015 [cited by applicant]
US 9524366B1 · Watanabe et al. · 2016 [cited by applicant]
US 9727438B2 · Buxbaum · 2017 [cited by applicant]
US 9811322B1 · Sindelar · 2017 [cited by applicant]
US 10871950B2 · De Smet · 2020 [cited by applicant]
US 10902121B2 · Chari · 2021 [cited by applicant]
US 11074107B1 · Nandakumar · 2021 [cited by applicant]
US 20040250157A1 · Barga et al. · 2004 [cited by applicant]
US 20090235252A1 · Weber et al. · 2009 [cited by applicant]
US 20140108438A1 · Duffy et al. · 2014 [cited by applicant]
US 20150222730A1 · Gower et al. · 2015 [cited by applicant]
US 20150293784A1 · Allen et al. · 2015 [cited by applicant]
US 20160103706A1 · Novaes · 2016 [cited by applicant]
US 20160117189A1 · Osborne et al. · 2016 [cited by applicant]
US 20180189389A1 · Baldini et al. · 2018 [cited by applicant]
US 20190073228A1 · Stanfill et al. · 2019 [cited by applicant]
US 20190140994A1 · Snider et al. · 2019 [cited by applicant]
US 20190229996A1 · Chofleming et al. · 2019 [cited by applicant]
US 20190378052A1 · Badhwar et al. · 2019 [cited by applicant]
US 20200265324A1 · Ferreira Moreno · 2020 [cited by examiner]
US 20210065048A1 · Salonidis · 2021 [cited by examiner]
US 20210118210A1 · Maruyama et al. · 2021 [cited by applicant]
US 20210216905A1 · Floratou et al. · 2021 [cited by applicant]
US 20220066905A1 · Lee et al. · 2022 [cited by applicant]
US 20220092471A1 · Jaeger et al. · 2022 [cited by applicant]
US 20220138266A1 · Wang et al. · 2022 [cited by applicant]
US 20220215260A1 · Dasoulas et al. · 2022 [cited by applicant]
US 20220374288A1 · Kibardin · 2022 [cited by examiner]
US 20230106416A1 · Gupte et al. · 2023 [cited by applicant]
US 20230168923A1 · Ganti et al. · 2023 [cited by applicant]
US 20230169408A1 · Andrade Costa et al. · 2023 [cited by applicant]
CN 104679807 · 2015 [cited by applicant]
CN 105117286A · 2015 [cited by applicant]
CN 105843684 · 2016 [cited by applicant]
CN 107273193A · 2017 [cited by applicant]
CN 109814986A · 2019 [cited by applicant]
CN 110275765A · 2019 [cited by applicant]
CN 110362315A · 2019 [cited by applicant]
CN 111158800A · 2020 [cited by applicant]
CN 111176817A · 2020 [cited by applicant]
CN 112379995A · 2021 [cited by applicant]
CN 112463346A · 2021 [cited by applicant]
Grafberger et al. Lightweight Inspection of Data Preprocessing in Native Machine Learning Pipelines, Jan. 2021 (Year: 2021). [cited by examiner]
Zhang et al., Diagnosing Machine Learning Pipelines with Fine-grained Lineage, 2017 (Year: 2017). [cited by examiner]
LIMA: Fine-grained Lineage Tracing and Reuse in Machine Learning Systems (Year: 2021). [cited by examiner]
List of IBM Patents or Applications Treated as Related, Nov. 2021. [cited by applicant]
Patel, Dhaval, et al. “Smart-ml: A system for machine learning model exploration using pipeline graph.” 2020 IEEE International Conference on Big Data (Big Data). IEEE, 2020. (Year: 2020). [cited by applicant]
Shang, et al. Democratizing Data Science Through Interactive Curation Of ML Pipelines, In Proceedings Of The International Conference On Management Of Data, Jun. 25, 2019, 18 pages. [cited by applicant]
Sparks, et al., KeystoneML: Optimizing Pipelines for Large-Scale Advanced Analytics, IEEE 33rd International Conference on Data Engineering (ICDE), Apr. 2017, pp. 535-546. [cited by applicant]
Betancourt, Frank, et al., “openDIEL: A Parallel Workflow Engine and Data Analytics Framework,” Proceedings of the Practice and Experience in Advanced Research Computing on Rise of the Machines (learning), 2019, pp. 1-7. [cited by applicant]
Nikitin, Nikolay O., et al., “Automated Evolutionary Approach for the Design of Composite Machine Learning Pipelines,” arXiv preprint arXiv:2106. 15397, 2021, 18 pages. [cited by applicant]