Skip to content

How custom Graph Convolutional Neural Networks helped a S&P500 company reach unprecedented Document AI performance

Read how you can easily handle up to 20K documents per month and how our custom solution reduced labelling time by ~6x in comparison to the performance of out-of-the-box Document AI.

About the customer

The customer is an S&P500 company that has a large real-estate portfolio, providing housing to more than 100.000 families.

The Problem / Challenge

During the course of a month, each property generates specific water, electricity, and gas bills across different vendors. Processing this information and making correct payments generates a tremendous amount of manual work. It is especially important that all bills are properly handled in time, as the single defect has potentially disastrous consequences (imagine having a family being unplugged from the grid).

In a competitive and rapidly growing environment, the S&P500 customer wanted to automate the processing of utility bills as much as possible, allowing them to process all bills on time, and within budget, while decreasing errors as much as possible.

In previous efforts, the customer experimented with Document AI and uptraining, which unfortunately did not lead to satisfactory results. This is why the S&P500 customer contacted us.

The Goal 

With the specific requirements of the customer, we had to consider developing a custom top-notch machine learning model that can deliver accurate predictions, scale to millions of bills per month, run using the fewest resources possible, and integrate with Document AI’s labelled document format.

The Solution

Devoteam G Cloud has a lot of experience building solutions based on the Document AI suite. To speed up development processes, we have created our own accelerator that automatically batches calls and optimises resource usage. The resulting architecture is depicted below:

It is battle-tested on a daily basis across several of our customers and is able to process hundreds of millions of documents per month.

For this project, we needed to accommodate a typical Document AI workflow in order to keep manual verification inside of the processes through Human-in-the-Loop (HITL). Therefore, our custom machine learning solution uses documents labelled through HITL as training data, and produces predictions that are compatible with the Document AI format. This requirement, in addition to bringing all of Document AI’s features available, makes us able to switch to any other Document AI processor if ever it eventually reaches a better performance than our custom models.

For the custom machine learning model, we opted for Graph Neural Networks, where each node represents a word in a document, and edges are links to the closest words in the document in the four cardinal directions. Each node contains several custom features, notably an embedding of the text as we did in previous works. Thus, this bases our solution on a language model, understanding text in a document, as well as on a structural model, understanding the relative positions of the texts in a document.

Finally, we use Vertex AI Vizier, Google Cloud’s black-box hyperparameter tuning service, allowing us to find the optimal configuration for our custom model in a matter of hours and reach the best possible performance.

The Methodology

As the S&P500 company deals with a large number of vendors, we had to take a phased approach. In fact, at the beginning of the project, labelling documents took more time to process the bills. To speed up the process, , we used active learning: the knowledge built from the model with a very low amount of training data can be leveraged to speed up further labelling, retrain the model and continue until we reach a satisfactory performance. The knowledge built on these vendors then allows us to concentrate efforts on other vendors and bootstrap them much faster.

As always, this machine learning project is mostly based on Google Cloud’s Vertex AI product suite. It allowed us to handle every step of our machine-learning project, from data exploration to model deployment and monitoring. Thanks to its capabilities, we were able to experiment with models and different architecture in a much faster way, allowing us to comply with all the requirements of the customer.

The Result

To evaluate the predictions of our created models, we use the F1 score metric. In a nutshell, the F1 score is made up of two parts: precision and recall. Precision is a measure of how accurate the predictions are. Recall is a measure of how complete the predictions are. To get the F1 score, we take the harmonic mean of the precision and recall scores. This helps evaluate how good the predictions are overall, because it takes into account both accuracy and completeness.

At the end of the project, thanks to Vertex AI Vizier which optimised our models’ parameters, we reached a 97.27% F1 score. Compared to the performance of Document AI, this means that the company spends ~6x less time labelling thanks to our custom solution.

Please note that an F1-score of 82.6% is already great and sufficient to run in production, but it can be further improved with custom machine learning models due to the specific challenges of that customer.

Model1-ECEPrecisionRecallF1 Score
DocumentAI66.87%84.75%80.56%82.60%
Custom model96.40%97.27%97.27%97.27%

Now that the solution is in production, the S&P500 company has onboarded more than 15 different vendors representing up to 20.000 documents per month. While the number of documents handled keeps growing, the team already benefits of the solution in their daily operations, making their workload much lighter, while reducing errors to almost zero.

Are you handling a high volume of documents manually?

Increase Operational Efficiency, Reduce Manual Effort & Accelerate Business Innovation with DocAI.