<!DOCTYPE article PUBLIC "-//NLM//DTD JATS (Z39.96) Journal Archiving and Interchange DTD v1.0 20120330//EN" "JATS-archivearticle1.dtd">
<article xmlns:xlink="http://www.w3.org/1999/xlink">
  <front>
    <journal-meta />
    <article-meta>
      <title-group>
        <article-title>Predictive Modelling of Trafic Accidents in Bogota, Colombia: Uncovering Key Contributing Factors</article-title>
      </title-group>
      <contrib-group>
        <contrib contrib-type="author">
          <string-name>Sebastián Castellanos</string-name>
        </contrib>
        <contrib contrib-type="author">
          <string-name>Alejandra Baena</string-name>
        </contrib>
        <contrib contrib-type="author">
          <string-name>Juan Camilo Ramírez</string-name>
        </contrib>
        <contrib contrib-type="author">
          <string-name>Universidad Antonio Nariño</string-name>
        </contrib>
        <contrib contrib-type="author">
          <string-name>Bogota</string-name>
        </contrib>
        <contrib contrib-type="author">
          <string-name>Colombia</string-name>
        </contrib>
      </contrib-group>
      <fpage>112</fpage>
      <lpage>117</lpage>
      <abstract>
        <p>Trafic accidents pose a significant threat to public safety, making their prediction and understanding of contributing factors crucial for efective preventive measures. This study focuses on leveraging historical accident data from Bogotá, Colombia, to design and evaluate machine learning models for trafic accident prediction and take the initial steps toward the identification of the most influential factors associated with each accident. The main objective of this research is to develop accurate machine learning models that can efectively predict trafic accidents in Bogotá and that can later be used in the identification of the key contributing factors leading to these incidents. By achieving this objective, it will be possible to enhance road safety and devise targeted interventions to reduce the occurrence of accidents in the city. A comprehensive dataset comprising historical trafic accident records in Bogotá was collected and preprocessed for analysis. Various machine learning algorithms, including decision trees, random forests, and neural networks, were applied to develop predictive models. The models were trained and evaluated using the F1-score as well as the area under the ROC curve. The experimental results demonstrate the efectiveness of machine learning models in predicting tracfi accidents in Bogotá. The best-performing models achieved performance scores over 0.80, both for the F1 metric and the area under the ROC curve, outperforming traditional statistical methods. These preliminary results are novel in the use of a more comprehensive and updated dataset of accidents in Bogotá and are envisaged to be extended with further analyses in order critical factors that strongly influence accident occurrence.</p>
      </abstract>
      <kwd-group>
        <kwd>eol&gt;Trafic accidents</kwd>
        <kwd>Machine Learning</kwd>
        <kwd>Prediction Models</kwd>
        <kwd>Contributing Factors</kwd>
        <kwd>Road Safety</kwd>
      </kwd-group>
    </article-meta>
  </front>
  <body>
    <sec id="sec-1">
      <title>1. Introduction</title>
      <p>
        The increase in road trafic accidents poses a significant challenge to urban areas worldwide,
affecting public safety, transportation eficiency, and overall societal well-being. The development
of efective strategies for accident prevention requires a deep understanding of the
contributing factors as well as an ability to accurately predict accident occurrences [
        <xref ref-type="bibr" rid="ref1 ref2 ref3">1, 2, 3</xref>
        ]. In recent
years, machine learning techniques have shown great potential in this domain by leveraging
historical accident data to identify patterns and extract valuable insights. The prediction of
trafic accidents using machine learning techniques has garnered significant attention globally
due to its potential to enhance road safety and inform efective accident prevention strategies.
Numerous studies have explored this area across various regions, aiming to develop accurate
models capable of forecasting accident occurrences [
        <xref ref-type="bibr" rid="ref4 ref5 ref6">4, 5, 6</xref>
        ]. Concurrently, a few investigations
have examined similar aspects within the context of Bogotá, Colombia [
        <xref ref-type="bibr" rid="ref10 ref11 ref7 ref8 ref9">7, 8, 9, 10, 11</xref>
        ]. Despite
these eforts, a notable gap persists in the current body of research: the absence of studies
leveraging the most recent and updated historical trafic accident data provided by the local
government of Bogotá.
      </p>
      <p>
        Bogotá, the capital city of Colombia, is home to a densely populated urban environment
characterized by complex trafic dynamics and diverse transportation modes. Understanding the
factors that contribute to trafic accidents in Bogotá is crucial for designing targeted interventions
and improving road safety initiatives. By harnessing the power of machine learning algorithms,
investigation in this line of research aims to develop accurate prediction models that can assist
policymakers, city planners, and law enforcement agencies in making informed decisions and
allocating resources efectively. Within the local context of Bogotá, previous studies have
made strides toward enhancing the understanding of trafic accident patterns and risk factors.
However, the majority of these investigations relied on historical data that may no longer
accurately reflect the evolving dynamics of the city’s trafic landscape [
        <xref ref-type="bibr" rid="ref10 ref11 ref7 ref8 ref9">7, 8, 9, 10, 11</xref>
        ]. Notably,
the local government of Bogotá has recently made available an updated repository of historical
trafic accident data, rendering previous analyses outdated and prompting the need for new
insights drawn from this comprehensive and up-to-date dataset.
      </p>
      <p>The present study builds upon a rich dataset comprising historical accident records collected
over the past decade in Bogotá. The dataset encompasses a wide range of variables, both temporal
and geographical, and accident severity. By systematically analyzing this comprehensive
dataset, we strive to identify the key factors influencing accident occurrences and evaluate the
performance of various machine learning models in predicting accidents with high precision
and recall rates.</p>
      <p>In conclusion, this article contributes to the growing body of research on using machine
learning for trafic accident prediction and factor identification. By applying advanced machine
learning techniques to historical accident data in Bogotá, we aim to enhance our understanding of
the factors influencing accident occurrences and provide valuable insights for policymakers and
stakeholders. The results of this study have the potential to inform evidence-based strategies for
improving road safety, reducing accident rates, and creating a safer transportation environment
in Bogotá and beyond.</p>
    </sec>
    <sec id="sec-2">
      <title>2. Related work</title>
      <p>Several studies have been conducted worldwide to leverage machine learning techniques for
trafic accident prediction and the identification of contributing factors. In this section, we
present a review of relevant literature, focusing on similar research eforts and their findings in
the context of trafic accident analysis.</p>
      <p>
        [
        <xref ref-type="bibr" rid="ref7">7</xref>
        ] consolidated a comprehensive dataset of motorcycle accidents in Colombia in the
20132018 period using various government sources. This dataset, including variables such as road and
weather conditions surrounding each incident, is proposed as the basis for future investigations
using predictive models in order to examine the main causes of trafic accidents involving
motorcycles. The same authors use this information in order to investigate motorcycle-related
accidents in Cartagena, Colombia, in order to identify areas within the city where the most
incidents occur, using a Bayes’ empirical approach, as well as contributing factors, such as the
number of intersections used by motorcyclists, which can be then used in order to implement
countermeasures [
        <xref ref-type="bibr" rid="ref8">8</xref>
        ].
      </p>
      <p>
        Following this line of research, [
        <xref ref-type="bibr" rid="ref9">9</xref>
        ] address the limited availability of suficient historical data
regarding trafic accidents in medium-size cities, such as Popayán, by employing complementary
data collection techniques, such as naturalistic driving, i.e., the continuous recording of driving
information in real-time. Prediction models trained on these data were found to exhibit
highperformance metrics and were used to identify regions within the urban area of the city where
accidents are concentrated. [
        <xref ref-type="bibr" rid="ref10">10</xref>
        ] conducted a similar study with accident data from 2016 in
Bogotá using multilayer perceptrons and naive Bayes models, finding that the former exhibit the
best performance and that the most contributing factor to trafic accidents is drivers’ behavior.
Finally, taking a diferent direction, [
        <xref ref-type="bibr" rid="ref11">11</xref>
        ] proposed the use of social media and meteorological
information as data sources resulting in high-performance models.
      </p>
      <p>While these studies above have contributed significantly to the field of trafic accident
prediction and factor identification, there is a limited number of studies focusing specifically
on Bogotá, Colombia. The present study aims to address this research gap by utilizing a
comprehensive dataset of historical accident records from Bogotá and applying a range of
machine learning models to predict accident occurrences accurately. Furthermore, our study
seeks to identify the most influential factors contributing to trafic accidents in the unique urban
context of Bogotá, providing valuable insights for policymakers, transportation authorities, and
urban planners in their eforts to improve road safety and reduce accident rates.</p>
    </sec>
    <sec id="sec-3">
      <title>3. Methodology</title>
      <p>Historical trafic accident data for Bogotá, Colombia, spanning the years 2015 to 2022, was
sourced from three distinct repositories managed by the local government. These repositories
contained detailed information on the vehicles involved, injured individuals, and fatalities
associated with each accident. The data, provided in CSV format, was merged into a consolidated
dataset based on a unique accident identification number shared across all three repositories.
The integrated dataset comprised over 220,000 instances, each encompassing the accident’s
temporal attributes (time, year, month, weekday), geographical coordinates (latitude, longitude),
accident severity (vehicle damage, injured, death), and the categorical class attribute "Type"
(crash, runover, fall from a vehicle). The dimensionality of the integrated dataset was streamlined
through the implementation of Principal Component Analysis (PCA). By selecting a subset
of factors that collectively explain no less than 95% of the variance, PCA is used to distill the
essential features while curtailing redundancy and noise.</p>
      <p>Three distinct Machine Learning models were selected for prediction: multilayer neural
networks, random forests, and decision trees. These models were chosen due to their capacity
to handle complex datasets and demonstrate proficiency in predictive tasks. Prior to training
the models, data preprocessing was carried out. This encompassed handling missing values,
encoding categorical variables, and normalizing numerical features to ensure consistency and
optimal model performance.
Model</p>
      <p>F1-Score</p>
      <p>Area Under ROC Curve
Multilayer Neural Network</p>
      <p>Random Forest
Support Vector Machine</p>
      <p>To robustly assess the models’ performance, a 10-fold cross-validation approach was employed.
The dataset was divided into ten subsets, with each model trained and evaluated ten times, using
a diferent subset as the validation set in each iteration while the rest were used for training.
The performance of the prediction models was evaluated using two key metrics: the F1 score
and the area under the Receiver Operating Characteristic (ROC) curve. The F1 score ofers
a balanced assessment of precision and recall, particularly relevant for imbalanced datasets
like this. The ROC curve and its associated area provide insights into the model’s ability to
discriminate between classes. Following the cross-validation process, the three models were
compared based on their F1 scores and ROC curve areas. This comparison aimed to identify
the model that demonstrated the most robust and accurate performance in predicting trafic
accidents and classifying their types.</p>
      <p>The study strictly adhered to ethical guidelines and data privacy regulations. The utilized data
was obtained from publicly available sources and did not contain any personally identifiable
information.</p>
    </sec>
    <sec id="sec-4">
      <title>4. Results</title>
      <p>All three prediction models exhibited a commendable level of performance, as evidenced by
F1-scores and AUC-ROC scores exceeding 0.80. This underscored their eficacy in efectively
predicting trafic accidents based on historical data. However, closer scrutiny of the results
revealed nuanced distinctions between the models. The multilayer neural network model
notably outperformed the others in terms of AUC-ROC score. This outcome signified its
superior ability to diferentiate between distinct accident types. Conversely, the support vector
machine model demonstrated a slight superiority in terms of F1-score, reflecting its capacity to
achieve a harmonious balance between precision and recall.</p>
      <p>The tabulated comparison is shown in Table 1 succinctly outlines the contrasting performance
of the three models in relation to both the F1-score and the area under the ROC curve. The
analysis reafirms the multilayer neural network’s superior AUC-ROC score, while the support
vector machine model’s marginally elevated F1-score showcases its prowess in achieving
precision-recall equilibrium.</p>
    </sec>
    <sec id="sec-5">
      <title>5. Conclusions</title>
      <p>This study addresses a significant gap in the field of trafic accident prediction by leveraging
advanced Machine Learning techniques to analyze the most recent and integrated historical
accident data provided by the local government of Bogotá. While previous research has explored
computational methods for investigating trafic accidents, our work stands out as the first to
employ an innovative approach that integrates multiple updated data repositories managed by
the city’s government.</p>
      <p>The outcomes of this study underscore the potential of machine learning methods in predicting
trafic accidents and shedding light on their underlying dynamics. By demonstrating the viability
of these approaches, we ofer compelling evidence that such predictive models can provide
crucial insights to enhance the local government’s ability to comprehend and address this
pressing issue. Our findings not only confirm the feasibility of utilizing machine learning in the
realm of trafic safety but also highlight the potential of these models to support the design and
implementation of preventive measures aimed at curbing this phenomenon.</p>
      <p>It is important to note that while our study presents promising preliminary results, there are
avenues for further exploration and refinement. Our current findings serve as a foundational
platform for future investigations, where a more comprehensive exploration of the most
influential factors contributing to trafic accidents can be undertaken. By integrating additional datasets
managed by the local government, such as weather and vehicle conditions, we anticipate that
these computational methods will provide even deeper insights into the complex interactions
surrounding trafic accidents.</p>
      <p>In conclusion, this study marks a significant step forward in the domain of trafic accident
prediction and prevention in Bogotá. The pioneering integration of diverse and updated datasets,
coupled with the application of machine learning models, has illuminated the potential to tackle
this challenge in a novel and efective manner. As we move forward, the insights gleaned from
this study will not only inform targeted interventions but also inspire continued research to
comprehensively understand and mitigate the causes and consequences of trafic accidents.</p>
    </sec>
    <sec id="sec-6">
      <title>Acknowledgments</title>
      <p>The authors would like to express their gratitude to Universidad Antonio Nariño1 for the
ifnancial support ofered during the completion of the present investigation.
1Universidad Antonio Nariño (https://www.uan.edu.co/).</p>
    </sec>
  </body>
  <back>
    <ref-list>
      <ref id="ref1">
        <mixed-citation>
          [1]
          <string-name>
            <given-names>F.-R.</given-names>
            <surname>Chang</surname>
          </string-name>
          ,
          <string-name>
            <given-names>H.-L.</given-names>
            <surname>Huang</surname>
          </string-name>
          ,
          <string-name>
            <given-names>D. C.</given-names>
            <surname>Schwebel</surname>
          </string-name>
          ,
          <string-name>
            <given-names>A. H. S.</given-names>
            <surname>Chan</surname>
          </string-name>
          ,
          <string-name>
            <given-names>G.-Q.</given-names>
            <surname>Hu</surname>
          </string-name>
          ,
          <article-title>Global road trafic injury statistics: Challenges, mechanisms and solutions</article-title>
          ,
          <source>Chinese journal of traumatology 23</source>
          (
          <year>2020</year>
          )
          <fpage>216</fpage>
          -
          <lpage>218</lpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref2">
        <mixed-citation>
          [2]
          <string-name>
            <given-names>A. A.</given-names>
            <surname>Mohammed</surname>
          </string-name>
          ,
          <string-name>
            <given-names>K.</given-names>
            <surname>Ambak</surname>
          </string-name>
          ,
          <string-name>
            <given-names>A. M.</given-names>
            <surname>Mosa</surname>
          </string-name>
          ,
          <string-name>
            <given-names>D.</given-names>
            <surname>Syamsunur</surname>
          </string-name>
          ,
          <article-title>A review of trafic accidents and related practices worldwide</article-title>
          ,
          <source>The Open Transportation Journal</source>
          <volume>13</volume>
          (
          <year>2019</year>
          ).
        </mixed-citation>
      </ref>
      <ref id="ref3">
        <mixed-citation>
          [3]
          <string-name>
            <given-names>L.</given-names>
            <surname>Wanumen</surname>
          </string-name>
          ,
          <string-name>
            <given-names>J.</given-names>
            <surname>Moreno</surname>
          </string-name>
          ,
          <string-name>
            <given-names>H.</given-names>
            <surname>Florez</surname>
          </string-name>
          ,
          <article-title>Mobile based approach for accident reporting</article-title>
          ,
          <source>in: Technology Trends: 4th International Conference, CITT</source>
          <year>2018</year>
          , Babahoyo, Ecuador,
          <source>August 29-31</source>
          ,
          <year>2018</year>
          ,
          <source>Revised Selected Papers 4</source>
          , Springer,
          <year>2019</year>
          , pp.
          <fpage>302</fpage>
          -
          <lpage>311</lpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref4">
        <mixed-citation>
          [4]
          <string-name>
            <given-names>T.</given-names>
            <surname>Bokaba</surname>
          </string-name>
          ,
          <string-name>
            <given-names>W.</given-names>
            <surname>Doorsamy</surname>
          </string-name>
          ,
          <string-name>
            <given-names>B. S.</given-names>
            <surname>Paul</surname>
          </string-name>
          ,
          <article-title>Comparative study of machine learning classifiers for modelling road trafic accidents</article-title>
          ,
          <source>Applied Sciences</source>
          <volume>12</volume>
          (
          <year>2022</year>
          )
          <fpage>828</fpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref5">
        <mixed-citation>
          [5]
          <string-name>
            <given-names>B. K.</given-names>
            <surname>Mohanta</surname>
          </string-name>
          ,
          <string-name>
            <given-names>D.</given-names>
            <surname>Jena</surname>
          </string-name>
          ,
          <string-name>
            <given-names>N.</given-names>
            <surname>Mohapatra</surname>
          </string-name>
          ,
          <string-name>
            <given-names>S.</given-names>
            <surname>Ramasubbareddy</surname>
          </string-name>
          ,
          <string-name>
            <given-names>B. S.</given-names>
            <surname>Rawal</surname>
          </string-name>
          ,
          <article-title>Machine learning based accident prediction in secure iot enable transportation system</article-title>
          ,
          <source>Journal of Intelligent &amp; Fuzzy Systems</source>
          <volume>42</volume>
          (
          <year>2022</year>
          )
          <fpage>713</fpage>
          -
          <lpage>725</lpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref6">
        <mixed-citation>
          [6]
          <string-name>
            <given-names>J.</given-names>
            <surname>Garcia</surname>
          </string-name>
          ,
          <string-name>
            <given-names>W.</given-names>
            <surname>Arias-Rojas</surname>
          </string-name>
          , G. Hernandez,
          <article-title>On the feasibility of real-time prediction of driver's trafic accident risk with auto ml using demographics and brain concentration levels information</article-title>
          ,
          <year>2022</year>
          , pp.
          <fpage>1</fpage>
          -
          <lpage>5</lpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref7">
        <mixed-citation>
          [7]
          <string-name>
            <given-names>H.</given-names>
            <surname>Ospina-Mateus</surname>
          </string-name>
          ,
          <string-name>
            <given-names>S. B.</given-names>
            <surname>Garcia</surname>
          </string-name>
          ,
          <string-name>
            <given-names>L. Q.</given-names>
            <surname>Jiménez</surname>
          </string-name>
          ,
          <string-name>
            <given-names>K.</given-names>
            <surname>Salas-Navarro</surname>
          </string-name>
          ,
          <article-title>Dataset of trafic accidents in motorcyclists in bogotá, colombia</article-title>
          ,
          <source>Data in brief 43</source>
          (
          <year>2022</year>
          )
          <fpage>108461</fpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref8">
        <mixed-citation>
          [8]
          <string-name>
            <given-names>H.</given-names>
            <surname>Ospina-Mateus</surname>
          </string-name>
          ,
          <string-name>
            <given-names>L. A. Q.</given-names>
            <surname>Jiménez</surname>
          </string-name>
          ,
          <string-name>
            <given-names>F. J.</given-names>
            <surname>Lopez-Valdes</surname>
          </string-name>
          ,
          <string-name>
            <given-names>S. S.</given-names>
            <surname>Sana</surname>
          </string-name>
          ,
          <article-title>Prediction of motorcyclist trafic crashes in cartagena (colombia): development of a safety performance function</article-title>
          ,
          <source>RAIRO-Operations Research</source>
          <volume>55</volume>
          (
          <year>2021</year>
          )
          <fpage>1257</fpage>
          -
          <lpage>1278</lpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref9">
        <mixed-citation>
          [9]
          <string-name>
            <given-names>S. F.</given-names>
            <surname>Yepes-Chamorro</surname>
          </string-name>
          ,
          <string-name>
            <given-names>J. J.</given-names>
            <surname>Paredes-Rosero</surname>
          </string-name>
          ,
          <string-name>
            <given-names>R.</given-names>
            <surname>Salazar-Cabrera</surname>
          </string-name>
          , Álvaro de la Cruz, J. M. Madrid-Molina,
          <article-title>Design, development and validation of an intelligent collision risk detection system to improve transportation safety: the case of the city of popayán, colombia</article-title>
          ,
          <source>Sustainability</source>
          <volume>14</volume>
          (
          <year>2022</year>
          )
          <fpage>10087</fpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref10">
        <mixed-citation>
          [10]
          <string-name>
            <given-names>H.</given-names>
            <surname>Vélez-Sánchez</surname>
          </string-name>
          ,
          <string-name>
            <given-names>H.</given-names>
            <surname>Saavedra-Angulo</surname>
          </string-name>
          ,
          <article-title>Use of data mining for root cause analysis of trafic accidents in colombia,</article-title>
          <year>2021</year>
          , pp.
          <fpage>674</fpage>
          -
          <lpage>688</lpage>
          .
        </mixed-citation>
      </ref>
      <ref id="ref11">
        <mixed-citation>
          [11]
          <string-name>
            <given-names>C.</given-names>
            <surname>Gutierrez-Osorio</surname>
          </string-name>
          ,
          <string-name>
            <given-names>F. A.</given-names>
            <surname>González</surname>
          </string-name>
          ,
          <string-name>
            <given-names>C. A.</given-names>
            <surname>Pedraza</surname>
          </string-name>
          ,
          <article-title>Deep learning ensemble model for the prediction of trafic accidents using social media data</article-title>
          ,
          <source>Computers</source>
          <volume>11</volume>
          (
          <year>2022</year>
          )
          <fpage>126</fpage>
          .
        </mixed-citation>
      </ref>
    </ref-list>
  </back>
</article>