Group 6 Team Members- Arjun Ghosh, Sujoy Banerjee, Saurabh Pal, Shaheryar Nadeem, Shubha Kotian
2024-12-10
document_id, year, document_reference, title, part, agency_parent_name, agency_name, restrictions, wordcount, shall, must, may_not, prohibited, required, restrictions_v2, sentence_length, conditionals_per_100_sentences, last_updated, long_word_score, acronyms_per_100_sentences, unique_acronymsScore to quantify compliance burden. Based on
this, a Compliance column was created to classify documents
as “Compliant” or “Non-Compliant.” The processed data, which served as
input to the ML model, looks like below:| document_id | year | document_reference | title | part | agency_parent_name | agency_name | restrictions | wordcount | shall | must | may_not | prohibited | required | restrictions_v2 | sentence_length | conditionals_per_100_sentences | last_updated | long_word_score | acronyms_per_100_sentences | unique_acronyms | Compliance |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 1.91e+10 | 2022 | Title 1, Part 1 | 1 | 1 | administrative committee of federal register | administrative committee of federal register | 0 | 256 | 0 | 0 | 0 | 0 | 0 | 0 | 25 | 10 | 1985-03-28 | 38.38362 | 0 | 0 | Non-Compliant |
shall, required, and
restrictions_v2 are closely related, highlighting their
joint influence on compliance. Conversely, weak or negative correlations
(blue) suggest minimal or inverse relationships, such as between
unique_acronyms and other features. This analysis helps
identify feature dependencies, reduce multicollinearity, and select
relevant predictors for model training.## | | | 0% | |======================================================================| 100%
## [1] "AUC: 0.999641414173515"
## [1] "MSE: 0.01138214445232"
The feature importance plot highlights the relative contributions of various features to the best-performing model’s predictions. Key insights include:
shall: The most influential feature,
indicating its strong predictive power in determining compliance.wordcount: A critical feature,
suggesting that longer documents are more likely to comply with
regulations.required: A significant feature,
highlighting its importance in compliance classification.restrictions, may_not, and
must have moderate influence, reflecting their partial
relevance to compliance prediction.acronyms_per_100_sentences,
part, and long_word_score show minimal
contributions, suggesting they play a negligible role in the model’s
decisions.This visualization provides transparency into how the model
prioritizes different features, helping validate the model’s alignment
with domain knowledge and guiding future feature engineering efforts.
The SHAP explanation plot provides insights into the contribution of individual features to the model’s prediction for a specific observation. Below is a summary of the key insights from the analysis:
required = 0:
required keyword has the most
significant negative impact on compliance likelihood.wordcount = 3:
shall = 0:
shall keyword negatively influences
the prediction, highlighting its importance in compliance
determination.restrictions, restrictions_v2, and
must:
prohibited, long_word_score, and
conditionals_per_100_sentences:
This SHAP analysis provides transparency into how the model arrives at its decision for this specific observation. Features with larger bars (positive or negative) have a stronger influence, while smaller bars indicate negligible impact. This ensures interpretability and aligns the model’s predictions with domain knowledge.
The Partial Dependence Profile (PDP) for the feature
sentence_length illustrates its relationship with the
average prediction made by the best-performing model. This graph helps
to understand how varying the values of sentence_length
influences the likelihood of the target outcome (e.g., compliance
vs. non-compliance) predicted by the model.
The H2O-based predictive model successfully categorized compliance costs with strong performance metrics, achieving a high AUC and low MSE on the test data. Key predictors, such as the frequency of terms like shall and required, along with document length, played a significant role in determining compliance.
The findings underscore the practical applicability of our model in regulatory environments: - Enhanced Compliance Strategies: Organizations can leverage the model to focus on impactful document features, thereby reducing risks and associated compliance costs. - Feature-Driven Insights: The focus on key predictors enables targeted improvements in documentation, aligning with regulatory expectations.